更新说明

README.md

@@ -31,9 +31,9 @@ YoloV7-tiny | https://github.com/bubbliiiing/yolov7-tiny-pytorch
 YoloV7-OBB | https://github.com/Egrt/yolov7-obb
 
 ## 性能情况
-| 训练数据集 | 权值文件名称 | 测试数据集 | 输入图片大小 | mAP 0.5 |
-| :-----: | :-----: | :------: | :------: | :------: |
-| SSDD | [yolov7_weights.pth](https://github.com/bubbliiiing/yolov7-pytorch/releases/download/v1.0/yolov7_weights.pth) | SSDD-Val | 640x640 | 85.4
+| 训练数据集 | 测试数据集 | 输入图片大小 | mAP 0.5 |
+| :-----: | :------: | :------: | :------: |
+| SSDD | SSDD-Val | 640x640 | 85.4
 ### 预测结果展示
 ![预测结果](img/test.jpg)
 ## 所需环境
@@ -172,4 +172,5 @@ img/street.jpg
 
 ## Reference
 https://github.com/WongKinYiu/yolov7
+
 https://github.com/bubbliiiing/yolov7-pytorch

yolo.py

@@ -25,7 +25,7 @@ class YOLO(object):
         #   验证集损失较低不代表mAP较高，仅代表该权值在验证集上泛化性能较好。
         #   如果出现shape不匹配，同时要注意训练时的model_path和classes_path参数的修改
         #--------------------------------------------------------------------------#
-        "model_path"        : 'model_data/ep100-loss0.022-val_loss0.034.pth',
+        "model_path"        : 'model_data/yolov7_obb_ssdd.pth',
         "classes_path"      : 'model_data/ssdd_classes.txt',
         #---------------------------------------------------------------------#
         #   anchors_path代表先验框对应的txt文件，一般不修改。
@@ -46,7 +46,7 @@ class YOLO(object):
         #---------------------------------------------------------------------#
         #   只有得分大于置信度的预测框会被保留下来
         #---------------------------------------------------------------------#
-        "confidence"        : 0.3,
+        "confidence"        : 0.5,
         #---------------------------------------------------------------------#
         #   非极大抑制所用到的nms_iou大小
         #---------------------------------------------------------------------#