|
@@ -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__
|