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
直接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