yolov5是一款非常好用到的目标检测算法，随着深度学习的发展，作者后续更新了分类和分割算法，yolov8甚至 加入了人体姿态估计算法，今天我们主要讲解一下分类算法！
最近正好看到了科大讯飞的比赛，里面用到了分类算法，想到yolov5有此功能就去尝试了一下！
官方仓库地址：https://github.com/ultralytics/yolov5
数据集操作：
数据集自己准备，这个是基础知识。不清楚的可以去简单学习一下。
我是在科大讯飞下载的数据集，只有一份训练集和比赛专用测试数据，需要自己手动分了训练集和验证集验证集。
我这里提供一个自己做的小脚本，自取，我是根据自己的需求写的，仅用于分类，谨慎食用
在yolov5-6.2文件同目录下创建datasets文件夹。
在datasets文件夹里存放着自己的数据集，例如，我得数据集文件夹 data 其格式如下
按标签分好类，验证集也是如此，我这个因为是比赛用的，所以不分测试集，最后到的数据上传科大讯飞官网评分。
2）开始训练
打开yolov5-3.2/classify/train.py，在yolov5环境下，运行。
修改：
--model', type=str, default='yolov5x-cls.pt'
--data', type=str, default=自己的数据地址
修改一个数据集的位置和一个权重文件，权重文件我会附在文末，说来这个好气，csdn上好多博主连这玩意都要收费，这不是欺负老实人吗
这样就跑成了
如果要测试精度，就自己在分一个测试集出来
----------------------------------------------------------------------------------
可能遇到的bug
AttributeError: partially initialized module ‘cv2‘ has no attribute ‘gapi_wip_gst_GStreamerPipeli
这个可能是CV2版本问题，重新装一下opencv
python调用git出错
#出错原因：git环境变量设置问题
#简便解决办法：在导入包的上方增加以下代码
import os
os.environ["GIT_PYTHON_REFRESH"] = "quiet"还有一个报错是报找不到图片，显示一堆的格式号，那个是因为数据集划分问题，比如你的图片分类做的有问题，train目录下是各个分类文件夹，同时还要有val和test至少一个，格式同train
分类权重文件
链接：https://pan.baidu.com/s/1U7TksPuZingHA9fmPoFqIw
提取码：0e72