update yolo.py

README.md

@@ -113,7 +113,7 @@ _defaults = {
     #   验证集损失较低不代表mAP较高，仅代表该权值在验证集上泛化性能较好。
     #   如果出现shape不匹配，同时要注意训练时的model_path和classes_path参数的修改
     #--------------------------------------------------------------------------#
-    "model_path"        : 'model_data/yolov5_s_v6.1.pth',
+    "model_path"        : 'model_data/yolov7_weights.pth',
     "classes_path"      : 'model_data/coco_classes.txt',
     #---------------------------------------------------------------------#
     #   anchors_path代表先验框对应的txt文件，一般不修改。
@@ -126,9 +126,11 @@ _defaults = {
     #---------------------------------------------------------------------#
     "input_shape"       : [640, 640],
     #------------------------------------------------------#
-    #   所使用的YoloV5的版本。s、m、l、x
+    #   所使用到的yolov7的版本，本仓库一共提供两个：
+    #   l : 对应yolov7
+    #   x : 对应yolov7_x
     #------------------------------------------------------#
-    "phi"               : 's',
+    "phi"               : 'l',
     #---------------------------------------------------------------------#
     #   只有得分大于置信度的预测框会被保留下来
     #---------------------------------------------------------------------#