diff --git a/README.md b/README.md index c23de9d5f315f7e8febcef2e6929e4b755acf31f..9286fbbd58d1c12aecc3ee95f5def4dcf385054d 100644 --- a/README.md +++ b/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 diff --git a/yolo.py b/yolo.py index 6777fc21d799f474b680afb13627713c70e6af56..28bb9ddfaaa6462ad59a8c8c9cf19d4b25b366aa 100644 --- a/yolo.py +++ b/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大小 #---------------------------------------------------------------------#