|
|
@@ -38,28 +38,20 @@ make all
|
|
|
└── testing/ # 单元测试目录
|
|
|
```
|
|
|
|
|
|
-## 更新模型
|
|
|
+## 模型配置
|
|
|
|
|
|
-目前训练过的模型：
|
|
|
+| 模型名称 | 模型配置 |
|
|
|
+| ------------------------------------------------------------------------------------------------------------------------------------------------ | ---------------------------------------------------------------------------------------- |
|
|
|
+| [Official YOLOv7](https://github.com/WongKinYiu/yolov7) | [./core/detectors/yolov7.py](./core/detectors/yolov7.py) |
|
|
|
+| [PaddleYOLO YOLOv7P6-E6](https://github.com/PaddlePaddle/PaddleYOLO/tree/develop/configs/yolov7#p6%E5%A4%A7%E5%B0%BA%E5%BA%A6%E6%A8%A1%E5%9E%8B) | [./core/detectors/paddle_yolo/\_\_init\_\_.py](./core/detectors/paddle_yolo/__init__.py) |
|
|
|
|
|
|
-- [Official YOLOv7](https://github.com/WongKinYiu/yolov7)
|
|
|
-- [PaddleYOLO YOLOv7P6-E6](https://github.com/PaddlePaddle/PaddleYOLO/tree/develop/configs/yolov7#p6%E5%A4%A7%E5%B0%BA%E5%BA%A6%E6%A8%A1%E5%9E%8B)
|
|
|
-
|
|
|
-更新模型时请指定模型路径和下载地址，如：
|
|
|
+如果更新了模型权重，请同时修改镜像中的下载地址，如：
|
|
|
|
|
|
```bash
|
|
|
-$ cat core/detectors/yolov7.py
|
|
|
-...
|
|
|
-PROJ_ROOT = Path(__file__).parent.parent.parent
|
|
|
-YOLO_DIR = str(PROJ_ROOT / "yolov7")
|
|
|
-# 模型权重路径
|
|
|
-WEIGHTS = str(PROJ_ROOT / "yiliv7_718.pt")
|
|
|
-...
|
|
|
-
|
|
|
$ cat Dockerfile
|
|
|
...
|
|
|
# 模型权重下载地址
|
|
|
-RUN wget ftp://192.168.199.31/yiliv7_718.pt
|
|
|
+RUN wget ftp://192.168.199.31/yolov7p6_e6_300e/20240314/yolov7p6_e6_300e_coco.zip
|
|
|
...
|
|
|
```
|
|
|
|
