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代表摄像头输入