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
转载声明:请注明本文章的标题及内容的出处和声明,谢谢
评论已关闭