leon il y a 1 mois
Parent
commit
ae8bda45f8
1 fichiers modifiés avec 1 ajouts et 0 suppressions
  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);
     
 };