1、前言
之前介绍了yolov5的目标检测示例,这次将介绍yolov5的分类展示
目标检测:YOLOv5 项目:训练代码和参数详细介绍(train)_yolov5训练代码的详解-CSDN博客
yolov5和其他网络的性能对比
yolov5分类的代码部分在这
2、数据集准备
yolov5分类的数据集就是常规的摆放方式
相同数据放在同样的目录下,目录名就是分类的类别
ClssifyDataSets├─train│ ├─目录1│ └─目录2├─val│ ├─目录1│ └─目录2
这里并非train、val、test都有,就类似于目标检测的数据,我们也没用到test