白嫖GPU之Google colab

前言：本人是一名穷且孤单的大学秃头生，跟着老师学习的是计算机视觉领域，学习目标检测到目标跟踪已经有一年的时间了，大多数都是在做论文阅读，很少有机会去实战（因为穷，买不起好的电脑，租不起服务器，实验室也不给资金哈哈哈）。直到近日我才偶然在某站上了解到了白嫖GPU的神器google colab，并且成功实现了yolov5口罩检测的训练（实测，很快，啊！）

写此博客以做记录和分享！

实现：YOLOv5实现口罩检测训练

数据集：百度网盘mask数据集

提取码：kw87

YoloV5：github地址

Google Chrome准备

我们所用的google colaboratory以及google drive均是源于谷歌旗下，所以我们需要先下载google Chrome浏览器（这里可以直接从官网下载，很简单，几乎傻瓜式安装即可）

这里我们需要下载谷歌官方的谷歌助手，或者是其他的谷歌浏览助手，这样我们才能够访问谷歌内部的网站，当然如果有些大佬强的话，可以直接搞一手国外的VPN，俗称FQ。

还有，大家要注册一个谷歌账号（后续步骤会用到）

这里可能会出现比较多的问题，但是这些问题都已经被解决了，所以大家可以直接询问度娘。

My Drive-Google Drive准备

假设大家已经完成了上一步。

Step1.这里我们打开谷歌自带的云盘，google-drive。

地址：My Drive

点击链接应该出现如下界面：

Step2.New-> Folder->Create

Step3.点击创建好的colab文件夹，右击上传文件，这里我们上传我们的数据集，口罩mask数据集（从百度网盘下载 解压至我们的桌面即可）

上传后应该如下图显示：

Step4.创建google colaboratory（这里需要先在浏览器中输入google colaboratory，连接到自己的google 硬盘 然后在google硬盘中刷新一下即可）

Google Colaboratory网址

Google Colaboratory实现

Step1：修改->笔记本设置->GPU

Step2:代码测试：

我们可以输入如下代码进行测试：

这里可以看出系统给出的是T4 GPU 对于学生党来说的我们可以说是很强的性能了，白嫖就很舒服。

输入如下代码后：

drive.mount(’/content/drive’)

会需要一个验证码，这里就需要我们注册的Google账号起作用了，进行授权获取之后 填写，就可以出现Mounted at /concent/drive结果

进行如下代码操作，实现目录更换，并将github上的yolov5项目克隆过来

由于google colab给的python里面 关于这个库的版本有问题，因此我们需要先将这个库进行升级，才能确保程序可运行。

–这里 由于我们训练的由原先的yolov5的80个类别变成了现在只有两个的类别（有口罩和无口罩），因此我们需要更改yolov5的配置文件。

这里我们回到My-drive中，发现在colab文件夹下多了yolov5文件夹

打开yolov5->models->yolov5s.yaml->右击open with text editor->修改里面的nc 从80->2->保存文件

回到colab中，运行如下内容：（运行成功，等待结束即可）