diff --git a/yolo.py b/yolo.py index 0962a78..f15011d 100644 --- 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,