yolov5训练_使用GoogleColab进行yolov5人脸口罩检测与识别模型训练

一、申请谷歌网盘

二、上传口罩数据集，网盘目录结构如下

三、申请Google Colaboratory应用

四、创建notebook

五、notebook增加GPU支持

查看显卡命令：

nvidia-smi

六、挂载网盘目录中的面具数据到当前运行环境下

并运行以下代码，中途需要验证：

import osfrom google.colab import drivedrive.mount('/content/drive')path = "/content/drive/My Drive"os.chdir(path)os.listdir(path)

google drive就会被挂载到目录/content/drive，后续就可以对google drive里的文件进行操作了

七、查看Pytorch和CUDA信息

框架信息

!pip3 listPackage                       Version        ----------------------------- ---------------absl-py                       0.10.0         alabaster                     0.7.12         albumentations                0.1.12         altair                        4.1.0          argon2-cffi                   20.1.0         asgiref                       3.3.0          astor                         0.8.1          astropy                       4.1            astunparse                    1.6.3          async-generator               1.10           atari-py                      0.2.6          atomicwrites                  1.4.0      dask                          2.12.0         dataclasses                   0.7            datascience                   0.10.6         Jinja2                        2.11.2         joblib                        0.17.0         jpeg4py                       0.1.4          ...  lmdb                          0.99           lucid                         0.3.8      pandas-datareader             0.9.0          pandas-gbq                    0.13.3        wordcloud                     1.5.0          wrapt                         1.12.1         xarray                        0.15.1         xgboost                       0.90           xkit                          0.0.0          xlrd                          1.1.0          xlwt                          1.3.0          yellowbrick                   0.9.1          zict                          2.0.0          zipp                          3.4.0          

八、下载yolov5框架源码

然后切换到google drive，修改yolov5/models/yolov5s.yaml，将原来的nc: 80改为nc: 2

九、开始训练模型

训练命令：

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

十、命令总结

# 查看显卡信息!nvidia-smi# 查看pip安装包!pip3 list# 查看Pytorch框架版本和cudnimport torchtorch.__version__# 挂载网盘到当前环境中import osfrom google.colab import drivedrive.mount('/content/drive')path = "/content/drive/My Drive"os.chdir(path)os.listdir(path)# 切换目录# 下载yolov5源码!git clone https://github.com/ultralytics/yolov5.git# 安装pyyaml依赖!pip install -U pyyaml# 切换到yolov5下载目录# 记得修改配置文件# 开始训练!python train.py --data ../mask/data.yaml --cfg models/yolov5s.yaml --weights '' --batch-size 32