1、ultralytics yolo图像分类训练案例
优点:使用方便,训练过程评估指标可以方便查看
缺点:自带模型少,可选择自定义小
参考:https://docs.ultralytics.com/tasks/classify/#val
https://blog.csdn.net/weixin_42357472/article/details/131412851
1)数据集格式
https://docs.ultralytics.com/datasets/classify/#dataset-format
2)训练:
cifar10数据集和yolov8n-cls.pt会默认下载,数据集默认会下载到/mnt/data/yolo/datasets/下,及自带可选模型,基于imagenet预训练的5个可选: