Google cloab上进行YOLOv5模型训练mask数据集

一.Google账号注册

手机下载qq邮箱，左下角语言选择英文，即可用国内手机号注册gmail邮箱。

二.Google云盘和Colab加载

云盘链接link

2.1Colab加载

按照上图可能加载不出来，没有Colab，
解决办法：
（1）设置里面，manage apps勾选【我的勾选了以后还是没有解决，用了方法2】
（2）利用网址link打开，编辑之后重新打开云盘就会出现了

2.2设置GPU环境

a)修改->笔记本设置->GPU
b)连接到托管代码执行程序
使用代码!nvidia-smi可以查看分配的GPU资源

2.3挂载云盘

将云盘资源挂载到Colab下就可以使用云盘的资源了
代码如下：

import os
from google.colab import drive`
drive.mount('/content/drive')

• 下面的代码作用

下面这一部分好像没有使用，还不太了解

path = "/content/drive/My Drive"

os.chdir(path)
os.listdir(path)`

运行上面的代码，弹出链接，点击链接，获取授权，把得到的验证码复制输入即可。

三.Colab常用操作

查看当前工作目录 pwd
查看当前目录中的内容%ls
进入My drive目录下%cd My\ drive
查看分配的资源!nvidia-smi

四.YOLOv5模型和mask数据集下载

YOLOv5模型

!git clone https://github.com/ultralytics/yolov5.git 

mask数据集
原始链接 https://public.roboflow.ai/object-detection/mask-wearing
百度网盘 https://pan.baidu.com/s/15GSPiJ59dg4kNyUch6W5Xw， 提取码：waj4

五.训练过程及问题

5.1训练过程

（1）重命名mask文件，将mask文件下的data.yaml用记事本打开，修改路径为

（2）修改yolov5->models->yolov5s.yaml文件：将nc=80改为nc=2，因为口罩数据集只有两种类别：mask，no-mask。
（3）pip install -r requirements.txt
安装依赖，尤其是遇到‘’thop‘’问题，安装即可解决
（4）利用cd命令进入yolovs下面
执行训练命令即可：

!python train.py --data ../mask/data.yaml --cfg models/yolov5s.yaml --weights '' --batch-size 64

（5）训练结束后会生成两个权重文件，best.pt和last.pt

(6)使用生成的best.pt权重文件进行测试：

!python detect.py --weight weights/best.pt --source ../mask/test/images/1288126-10255706714jpg_jpg.rf.95f7324cbfd48e0386e0660b5e932223.jpg

测试结果：

5.2问题与思考

（1）未修改mask文件下的data.yaml路径，导致报错

（2）cd命令进入yolov5目录中

解决办法：采用%ls查看文件名称，更改正确路径，但是还是不知道为什么会变成’yolov5(1)’，可能因为我clone了两次yolov5，导致文件路径名改变了

（3）出现错误代码：

解决办法：在train.py中删除sort_keys相关的代码即可（原因不知，待解决）
（4）在本地运行训练时，出现Broken pipe，应该是网速问题导致。在Colab上未出现该error

5.3下一步目标

学习R-CNN，YOLOv5网络层结构，调参感受检测效果，了解原理，了解代码参数，训练自己的数据集。

参考

【迷途小书童】：使用GooleColab训练yolov5模型
https://xugaoxiang.com/2020/11/01/google-colab-yolov5/
https://www.bilibili.com/video/BV1YK4y1E7zd
【迷途小书童】YOLOv5模型训练
https://xugaoxiang.com/2020/07/02/yolov5-training/