mirror of
https://github.com/Karllzy/cotton_color.git
synced 2025-11-08 18:53:53 +00:00
Merge remote-tracking branch 'origin/YOLO' into YOLO
This commit is contained in:
commit
47318eff64
@ -1,6 +1,10 @@
|
||||
# YOLO调试记录
|
||||
|
||||
## data 数据yaml文件配置
|
||||
|
||||
## data文件夹 yaml文件配置
|
||||
|
||||
该文件夹中配置的是数据集的路径及类别信息
|
||||
|
||||
|
||||
```python
|
||||
path: ../datasets/dimo4
|
||||
@ -20,9 +24,12 @@ names:
|
||||
python export.py --weights runs/train/exp4/weights/best.pt --img 640 --batch 1 --device 0 --include onnx --opset 15
|
||||
```
|
||||
|
||||
weights地址: runs/train/exp4/weights/xxx.pt 路径中的模型为pt文件
|
||||
|
||||
在detect.py中batch为1 batch在detect中不能更改
|
||||
通过export.py导出的文件格式为onnx
|
||||
|
||||
weights地址: runs/train/exp4/weights/xxx.pt best.pt为train.py训练后所得模型
|
||||
|
||||
在detect.py中batch为1,batch在detect中不能更改,所以在导出onnx模型文件时要设置batch 1 batch在detect中不能更改
|
||||
|
||||

|
||||
|
||||
@ -45,15 +52,31 @@ onnx
|
||||
```
|
||||
|
||||
|
||||

|
||||
|
||||
## 模型加载
|
||||
|
||||
### 命令行形式
|
||||
|
||||
```bash
|
||||
python detect.py --weights runs\train\exp4\weights\best.onnx --data D:\yolov5- master\data\dimo4.yaml
|
||||
```
|
||||
|
||||

|
||||
|
||||

|
||||
|
||||

|
||||
|
||||
### PyCharm配置
|
||||
|
||||
```bash
|
||||
--weights
|
||||
runs\train\exp4\weights\best.onnx
|
||||
--data
|
||||
D:\yolov5-master\data\dimo4.yaml
|
||||
```
|
||||

|
||||
|
||||
在detect.py中
|
||||
|
||||
@ -61,8 +84,10 @@ D:\yolov5-master\data\dimo4.yaml
|
||||
|
||||
或者 :runs/train/exp4/weights/xxx.onnx
|
||||
|
||||
|
||||

|
||||
|
||||

|
||||
|
||||

|
||||

|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user