leon 1 tháng trước cách đây
mục cha
commit
18e4409e58
1 tập tin đã thay đổi với 3 bổ sung0 xóa
  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);