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

+ 3 - 0
src/infer/trt/depth/depth.hpp

@@ -58,6 +58,9 @@ namespace depth
 
             output_buffer_.gpu(input_numel / 3);
             output_buffer_.cpu(input_numel / 3);
+
+            affine_matrix_.gpu(6);
+            affine_matrix_.cpu(6);
         }
     
         void preprocess(const tensor::Image &image, affine::LetterBoxMatrix &affine, void *stream = nullptr);