|
|
@@ -56,11 +56,11 @@ namespace yolo
|
|
|
|
|
|
bool load(const std::string &engine_file, ModelType model_type, 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) override;
|
|
|
+ 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);
|
|
|
|
|
|
- virtual data::BoxArray forward(const tensor::Image &image, void *stream = nullptr) override;
|
|
|
+ virtual data::BoxArray forward(const tensor::Image &image, void *stream = nullptr);
|
|
|
|
|
|
- virtual data::BoxArray forwards(void *stream = nullptr) override;
|
|
|
+ virtual data::BoxArray forwards(void *stream = nullptr);
|
|
|
};
|
|
|
|
|
|
Infer *loadraw(const std::string &engine_file, ModelType model_type, float confidence_threshold,
|
