leon 1 개월 전
부모
커밋
3fae9a59ac
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      src/infer/trt/depth/depth.cu

+ 1 - 1
src/infer/trt/depth/depth.cu

@@ -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();
 }