bugfix

src/infer/trt/depth/depth.cu

@@ -142,10 +142,9 @@ Result DepthModelImpl::forward(const tensor::Image &image, void *stream)
 }
 
 
-Infer *loadraw(const std::string &engine_file, ModelType model_type, const std::vector<std::string>& names, float confidence_threshold,
-    float nms_threshold) 
+Infer *loadraw(const std::string &engine_file) 
 {
-    DepthModelImpl *impl = new YoloModelImpl();
+    DepthModelImpl *impl = new DepthModelImpl();
     if (!impl->load(engine_file)) 
     {
         delete impl;

src/infer/trt/depth/depth.hpp

@@ -61,7 +61,7 @@ namespace depth
         }
     
         void preprocess(const tensor::Image &image, affine::LetterBoxMatrix &affine, void *stream = nullptr);
-        void postprocess(int width, int height, void *stream = nullptr)
+        void postprocess(int width, int height, void *stream = nullptr);
         
     
         bool load(const std::string &engine_file);
@@ -70,8 +70,7 @@ namespace depth
     
 };
 
-Infer *loadraw(const std::string &engine_file, ModelType model_type, const std::vector<std::string>& names, float confidence_threshold,
-    float nms_threshold);
+Infer *loadraw(const std::string &engine_file);
 
 std::shared_ptr<Infer> load_depth(const std::string &engine_file, int gpu_id);