Răsfoiți Sursa

fix bug for yolov3 mobilenetv3

jiangjiajun 5 ani în urmă
părinte
comite
84b0e6e86b
1 a modificat fișierele cu 2 adăugiri și 2 ștergeri
  1. 2 2
      paddlex/cv/models/yolo_v3.py

+ 2 - 2
paddlex/cv/models/yolo_v3.py

@@ -95,9 +95,9 @@ class YOLOv3(BaseAPI):
         elif backbone_name == 'MobileNetV1':
             backbone = paddlex.cv.nets.MobileNetV1(norm_type='sync_bn')
         elif backbone_name.startswith('MobileNetV3'):
-            models_name = backbone_name.split('_')[1]
+            model_name = backbone_name.split('_')[1]
             backbone = paddlex.cv.nets.MobileNetV3(
-                norm_type='sync_bn', models_name=models_name)
+                norm_type='sync_bn', model_name=model_name)
         return backbone
 
     def build_net(self, mode='train'):