YOLOV5训练部署

1.标图

(1).下载labelImg标图工具

(2).数据图片放在一个文件夹内进行标注

2.制作yolo数据格式

(1)Z:\Detect\Data\tmp路径下data.zip解压缩

(2)标定数据复制到对应路径制作voc

(3)运行Z:\Detect\voc2coco\voc2yolo.py脚本

3.训练

配置voc_DebugMK.yaml;数据路径、类别、类别名称

train.py --data data/voc_DebugMK.yaml --cfg models/yolov5s_DebugMK.yaml --weights weights/yolov5s.pt --batch-size 8 --epochs 50

4.测试

配置detect.py VSCode启动参数

5.模型转换

(1)onnx

YOLOV5训练部署_第1张图片

 直接vscode运行改文件;或yolov5文件夹路径下命令行运行该文件

(2)Openvino

a.

pip install --upgrade defusedxml -i Simple Index

pip install --upgrade networkx -i Simple Index

b.管理员权限运行

python mo_onnx.py --input_model Z:/Detect/yolov5/weights/best.onnx

你可能感兴趣的:(深度学习模型训练,深度学习)