Yolov5 安装和使用

Yolov5 安装和使用（机器学习笔记）

特别说明

本文不涉及算法等内容，仅是使用。

视频

我也是从这里看到素材准备和训练的过程
B站：https://www.bilibili.com/video/BV1dF411i7d7

yolov5

仓库
https://github.com/ultralytics/yolov5

安装引用
pip install -r requirements.txt

数据标记工具 labelImg

https://github.com/heartexlabs/labelImg

启动（切换为 yolo 模式）
.\labelImg 文件夹路径 标签文本
例如：
labelImg .\images\ .\tag.txt

数据训练

分类
python classify.py

训练
python train.py --data ./test/test.yaml --weights yolov5s.pt --img 640 --epochs 200 --batch-size 16

测试图片
python detect.py --weights ./best.pt --source ./223.png

测试视频
python detect.py --weights ./best.pt --source ./223.mp4 --view-img

模型转换

python export.py --weights yolov5s.pt --img 640 --batch 1 --opset 9 --optimize --simplify

安装驱动
# CPU
pip install -r requirements.txt coremltools onnx onnx-simplifier onnxruntime openvino-dev tensorflow-cpu
# GPU
pip install -r requirements.txt coremltools onnx onnx-simplifier onnxruntime-gpu openvino-dev tensorflow

指定格式 onnx
python export.py --weights yolov5s.pt --img 640 --batch 1 --opset 9 --optimize --simplify --include onnx
python export.py --weights yolov5m6.pt --img 1280 --batch 1 --opset 9 --optimize --simplify --include onnx

---

版权属于：zgcwkj

本文链接：https://blog.zgcwkj.cn/archives/212.html

转载声明：请注明本文章的标题及内容的出处和声明，谢谢