leon 1 tháng trước cách đây
mục cha
commit
6d08abab79
1 tập tin đã thay đổi với 2 bổ sung2 xóa
  1. 2 2
      src/infer/infer.hpp

+ 2 - 2
src/infer/infer.hpp

@@ -17,8 +17,8 @@ class Infer{
 public:
     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) = 0;
     virtual data::BoxArray forward(const tensor::Image &image, void *stream = nullptr) = 0;
-}
+};
 
-std::shared_ptr<Infer> load(const std::string& model_path, ModelType model_type, const std::vector<std::string>& names={}, int gpu_id=0, float confidence_threshold=0.5f, float nms_threshold=0.45f);
+std::shared_ptr<Infer> load(const std::string& model_path, ModelType model_type, const std::vector<std::string>& names, int gpu_id=0, float confidence_threshold=0.5f, float nms_threshold=0.45f);
 
 #endif // INFER_HPP__