bugfix

src/infer/trt/yolo.hpp

@@ -56,7 +56,7 @@ namespace yolo
         void preprocess(int ibatch, affine::LetterBoxMatrix &affine, void *stream = nullptr);
         
     
-        bool load(const std::string &engine_file, ModelType model_type, float confidence_threshold, float nms_threshold);
+        bool load(const std::string &engine_file, ModelType model_type, const std::vector<std::string>& names, float confidence_threshold, float nms_threshold);
     
         virtual data::BoxArray forward(const tensor::Image &image, int slice_width, int slice_height, float overlap_width_ratio, float overlap_height_ratio, void *stream = nullptr);