自学内容网 自学内容网

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)!