Update yolo.py

bubbliiiing · Nov 30, 2020 · de18de8 · de18de8
1 parent b0cb4c4
commit de18de8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/yolo.py b/yolo.py
@@ -24,10 +24,10 @@
 #--------------------------------------------#
 class YOLO(object):
     _defaults = {
-        "model_path"        : 'model_data/yolov4_mobilenet_v3_map75.13.pth',
+        "model_path"        : 'model_data/yolov4_mobilenet_v2_map76.93.pth',
         "anchors_path"      : 'model_data/yolo_anchors.txt',
         "classes_path"      : 'model_data/voc_classes.txt',
-        "backbone"          : 'mobilenetv3',
+        "backbone"          : 'mobilenetv2',
         "model_image_size"  : (416, 416, 3),
         "confidence"        : 0.5,
         "iou"               : 0.3,