|
|
@@ -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);
|
|
|
|
