若该文为原创文章，转载请注明原文出处。
一般情况下，大部分人的电脑都是没有cpu的，cpu也是可以训练的，但花费的时间太长，实际200张图片，使用CPU训练300轮花了3天，本章记录使用云服务器来训练自己的数据集。
使用的云服务器是AutoDL,一直在使用，性价比还是比较高的。
AutoDL算力云 | 弹性、好用、省钱。租GPU就上AutoDL
训练疲劳驾驶行为检测
一、准备数据集
数据集可以自己拍照，也可以网上找，博主在网上找到了一份疲劳驾驶行为的图片，直接拿过来用了，共有2千多张。
准备好数据后需要数据标注，数据标注是大部分人工智能算法得以有效运行的关键环节。数据标注的过程是通过人工贴标的方式，为机器系统可供学习的样本。yolov5标注使用的是labelimg软件。
需要数据集，下方评论或@博主。标注数据是在本地电脑操作，标注后才把数据上传到云服务器训练
二、labelimg
1、激活环境
conda activate yolo
编辑
2、安装labelimg
# 使用pip 安装 pip install labelimg
安装后，在终端输入labelimg启动软件
3、标注
标记完成的数据请按照下面的格式进行放置，方便程序进行索引。 colo128 ├─ images │ ├─ test # 下面放测试集图片 │ ├─ train # 下面放训练集图片 │ └─ val # 下面放验证集图片 └─ labels ├─ test # 下面放测试集标签 ├─ train # 下面放训练集标签 ├─ val # 下面放验证集标签
3.1 打开图片及设置标注文件保存的目录并设置自动保存，这里需要主要选择yolo格式
3.2 开始标注，画框，标记目标的label，crtl+s保存，然后d切换到下一张继续标注，不断重复重复
标注完成后，把colo128打包压缩下，准备上传到服务器。
三、训练数据集
1、注册AutoDL
自行注册，云服务可以按时收费，也可以其他方式，自己选择。
2、创建实例
配置选择，选择的是3090，也可以其他，我比较喜欢就是直接选择好社区镜像，就给你创建好环境
创建后开启，可以通过多种方式登录，为了方便，直接使用JupyterLab
JupyterLab，打开终端就可以看到终端界面了，环境也安装好了
3、上传数据集和下载YOLOV5-5.0
删除环境自带的yolov5版本，然后下载yolov5-5.0版本，并上传到服务器，可以直接拖进去，并把标注好的数据也一并上传。
4、修改配置文件
博主是 基于colo128配置文件训练的，这里需要修改两个文件
修改文件一：复制yolov5-5.0/data/colo128.yaml为pilao_coco128.yaml
# 修改一：数据集路径 train: ../coco128/images/train2017/ # 128 images val: ../coco128/images/train2017/ # 128 images # number of classes # 修改，原本是80,修改成3 nc: 3 # 修改二：class names即标注的类,数据集标注了3类，所以修改成3类，colo128是80类 # class names names: [ 'closed_eye','closed_mouth','open_eye']
修改文件二：复制yolov5-5.0/models/yolov5s.yaml为pilao_yolov5s.yaml
注意不一定是yolov5s.yaml，这是训练时参考的模型，可以是其他的，自行了解
# parameters # 修改，原本是80,修改成3 nc: 3 # number of classes depth_multiple: 0.33 # model depth multiple width_multiple: 0.50 # layer channel multiple
5、训练
python train.py --img 640 --batch 16 --epochs 300 --data ./data/pilao_coco128.yaml --cfg ./models/pilao_yolov5s.yaml
看到进度，已经在运行了，训练了大概3个多小时。
最后可以在runs/train/expX/weights目录下看到生成了两个pt文件
6、测试
python detect.py --source data/images/zidane.jpg --weights runs/train/exp2/weights/best.pt
路径自行修改。文章来源：https://www.toymoban.com/news/detail-617911.html
如有侵权，或需要完整代码，请及时联系博主。文章来源地址https://www.toymoban.com/news/detail-617911.html
到了这里，关于AI学习笔记四：yolov5训练自己的数据集的文章就介绍完了。如果您还想了解更多内容，请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章，希望大家以后多多支持TOY模板网！
