Google Colab的详细使用教程—YOLOv3为例

一、先哔哔几句热下身

相信疫情隔离在家而且还要使用深度学习的小伙伴们或多或少都会面临么的钱买高性能显卡又么的钱租服务器的窘境。下面我就来给小伙伴们介绍以下如何使用Google Colab免费服务器的故事吧（至于怎么上Google大家自行解决哦）。

Google Colaboratory是谷歌提供的基于linux系统的免费云平台，内部已经集成了深度学习所需要的库，比如Tensorflow（目前Version：2.2.0rc3）和Keras（目前Version: 2.3.1）等，一般情况下无需我们再做配置，直接上传自己代码就能用了。至于一些深度学习框架诸如Tensorflow的版本可能与自己代码不匹配的问题，后面也会讲解如何解决，下面就开始快乐的编程之旅吧~。

二、如何进入Colab——故事得从Google云端硬盘讲起

为了方便以后代码文件的管理，我们先介绍下 Google云端硬盘吧。后面推荐大家使用 Google云端硬盘上传文件然后加载到自己的Colab里。

首先你得注册个Google账号-->Google搜索“Google 云端硬盘”-->个人 转至Google云端硬盘，然后你就可以看到以下界面啦：

 

 

大家可以新建属于自己的Colab和文件夹，新建的Colab的默认名称为Untitled，代码运行就在这里面操作，后面也以这个为例讲解。不过现在莫要着急嘛，咱先把自己的工程代码上传到个人的文件夹中，这里我的叫test（我上传到test里的文件夹叫yolo3-keras，后面的栗子会用到）。步骤为：双击test文件夹-->进去后右键-->上传文件/上传文件夹-->根据需要上传自己的代码文件或文件夹，上传时间长短视网络情况而定。完成这一步之后，我们就可以开始快乐的操作Colab啦~

三、Colab如何操作？——莫慌，故事不长，听我慢慢道来

在步骤二的基础上，我们双击自己建立的Colab，进去之后它长这样：

具体一些区域的功能已经在图上注释了，大家可以先了解下。首先当然是先入门hello world啦~如下所示，在代码框里输入print("hello world!")，然后点击左边的运行按钮就可以啦~~

那嘛现在默认大家已经入门Colab的使用了哦，下面开始讲解如何运行自己在步骤二中上传的代码工程的操作（以训练yolov3网络为例）。

首先是设置加速器为GPU加速，大家可以在工具栏找到修改-->笔记本设置-->硬件加速器-->选择GPU，这样就可以在后面使用GPU加速啦。

下一步就是装载自己的Google云端硬盘，用以后面加载我们放在云盘中的文件。方法：侧边栏-->文件-->装载Google云端硬盘。见下图：

装载完成之后的界面如下： 

加载完之后就开始导入自己在步骤二中存入硬盘的代码文件吧~ 这里使用的命令行基本与Linux命令一样，有一点不同是需要在命令前加上“!”，首先把自己的工程文件拷贝到/content文件夹下，（查找文件的路径的快捷方法可以在相关文件夹下右键-->复制路径）我这里输入的指令为：!cp -r "/content/drive/My Drive/test/yolo3-keras" "/content/"，当然需要根据各自的情况来输入路径。结果如下：

OK，到这里就可以使用指令开心地运行自己的代码啦~比如我要运行我yolo3-keras里的网络训练文件train.py，则我只用输入命令：!python /content/yolo3-keras/train.py就OK啦~~~如下：

哦豁，看上图就会发现这里出问题咯。这是咋回事呐 ?这是由于Colab自带的Tensorflow版本(Version：2.2.0rc3)与我们工程里的版本(我用的V1系列的)不匹配导致的，可以通过命令：!pip show tensorflow来查看当前的版本，当然，查看其他module版本的方法类似。这问题可咋办呐？解决方法很简单-->卸了安装符合自己版本的tensorflow就可以啦。我们可以使用命令：

!pip uninstall tensorflow来卸载原来的版本，然后使用命令：!pip install tensorflow==1.13.2来安装符合自己要求的版本啦，比如我这里装的是1.13.2版本的。由于该步骤操作简单，就不上图啦~

下一个就是训练完成后如何讲自己训练的权重文件下载到本地呐？方法有很多种：可以直接在相应的文件上右键-->下载，也可以使用!cp指令放到自己的云端硬盘里，然后再下载等等，大家各取所需哦~

四、解决Colab长时间无操作就掉线的问题

大家也会发现，当我们在训练自己的模型的时候，如果长时间不操作Colab后他就会自动断线，这就很头疼了，我们也不能傻傻一直盯着它跑吧。莫慌，你可以通过以下简单的操作来解决：

参考：https://medium.com/@shivamrawat_756/how-to-prevent-google-colab-from-disconnecting-717b88a128c0

首先在Colab界面打开控制台（快捷键Ctrl+Shift+I），然后复制以下代码（这段代码的功能是设置每隔60000ms也就是1min自动点击一下Colab的“连接”操作，这样就不至于长时间误操作而导致自动断开连接啦，当然这只是个例子，方法不唯一，大家也可以调整下自己的间隔时间）：

function ClickConnect(){
    console.log("Working");
    document.querySelector("colab-toolbar-button#connect").click()
}
setInterval(ClickConnect, 60000)

把上面这段代码添加到下图所示的位置 ，然后敲一下回车键，OK，开始快乐编程之旅吧~~~

（我电脑上用这种方法貌似效果不咋好，不知道其他人的咋样）

 

好啦，目前为止Colab的使用方法就差不多讲完啦，当然这只是一些常用方法，其他操作可以在这些的基础上举一反三哦，我也是新手，Colab还有很多功能等着大家去发现呐！