From dbdd88776b7f0bfcc47fcf4a1a7511c93a1ae408 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=99=BD=E9=B9=AD=E5=85=88=E7=94=9F?= <766529835@qq.com>
Date: Mon, 6 Feb 2023 01:24:10 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=E8=AF=B4=E6=98=8E?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 README.md | 7 ++++---
 yolo.py   | 4 ++--
 2 files changed, 6 insertions(+), 5 deletions(-)

diff --git a/README.md b/README.md
index c23de9d..9286fbb 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 6777fc2..28bb9dd 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大小
         #---------------------------------------------------------------------#
-- 
GitLab