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);