|
@@ -151,7 +151,7 @@ Result DepthModelImpl::forward(const tensor::Image &image, void *stream)
|
|
|
depth_map_buffer_.gpu_bytes(), cudaMemcpyDeviceToHost, stream_));
|
|
|
checkRuntime(cudaStreamSynchronize(stream_));
|
|
|
cv::Mat depth_mat(image.height, image.width, CV_32FC1, depth_map_buffer_.cpu());
|
|
|
- return depth_mat;
|
|
|
+ return depth_mat.clone();
|
|
|
}
|
|
|
|
|
|
|