|
|
@@ -546,7 +546,7 @@ Infer *loadraw(const std::string &engine_file, ModelType model_type, float confi
|
|
|
float nms_threshold)
|
|
|
{
|
|
|
YoloModelImpl *impl = new YoloModelImpl();
|
|
|
- if (!impl->load(engine_file, yolo_type, confidence_threshold, nms_threshold))
|
|
|
+ if (!impl->load(engine_file, model_type, confidence_threshold, nms_threshold))
|
|
|
{
|
|
|
delete impl;
|
|
|
impl = nullptr;
|
