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

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

@@ -67,6 +67,7 @@ namespace depth
         bool load(const std::string &engine_file);
     
         virtual Result forward(const tensor::Image &image, void *stream = nullptr);
+        virtual Result forward(const tensor::Image &image, int slice_width, int slice_height, float overlap_width_ratio, float overlap_height_ratio, void *stream = nullptr);
     
 };