yolo5快速上手训练数据集
修改训练类目数量:
根目录新建datasets目录
names配置,就是lables中的类。有几个写几个
训练命令案例:
不带预训练权重
python .\train.py --data .\data\coco_number.yaml --cfg .\models\yolo5s_number.yaml --batch-size 20 --epochs 10
带权重
python .\train.py --data .\data\coco_number.yaml --cfg .\models\yolo5s_number.yaml --weights .\runs\train\exp11\weights\best.pt --batch-size 20 --epochs 10
训练好得到pt权重,可以直接利用现有的代码进行推理
0代表摄像头输入
原文地址:https://blog.csdn.net/weixin_52236586/article/details/142874387
免责声明:本站文章内容转载自网络资源,如本站内容侵犯了原著者的合法权益,可联系本站删除。更多内容请关注自学内容网(zxcms.com)!