YOLOv5 图片分类

 官网链接、

1、命令

1）没有模型配置，只能通过--model配置加载预训练模型

python train.py --epochs 5 --batch-size 4 --workers 4 --img 224 --data E:\数据集\flower_photos\flower_data --model ./checkpoints/yolov5s-cls.pt 

2、输出分类类别修改

1）修改部位

reshape_classifier_output(model, nc)  # update class count

2）需要修改最后一个线性分类器的输出类别个数