使用colab训练GitHub中的模型——以tensorflow/models为例

使用colab训练GitHub中的模型——以tensorflow/models为例

没有显卡如何训练GitHub中的模型。

1. colab简介

colab全名colaboratory，是google推出的在线代码编辑器，其实就是个在线jupyter笔记本环境，可以免费提供强劲的GPU算力。
colab的使用和google drive是分不开的。可以这样理解，google drive就是硬盘，可以存代码和数据，colab是jupyter笔记本，通过Python或者命令行代码块可以调用硬盘中的代码文件或者直接编写代码段，代码文件可以从硬盘中调用数据。colab文件本身也是一个.ipynb文件，存在google drive之中。
使用colab的前提如下：

1. 有一个google账号
2. 连接到google drive

这两个前提需要科学上网。

2. 新建colab笔记本

在google drive中右键->更多->关联更多应用，搜索colaboratory添加即可（下图是我已经添加过的，第一次使用需要关联）。关联之后google drive中右键->更多->google colaboratory即可创建colab笔记本。

下图为colab笔记本界面。笔记本分为代码块和文本块。单击代码块可以编辑代码，左侧会有一个运行代码按钮。

3. 模型训练

这里我使用的模型为目标检测模型，GitHub地址。

3.1 挂载谷歌网盘

需要将colab和google drive关联起来，后续会将从GitHub下载的模型以及其它数据下载到google drive中。

from google.colab import drive
drive.mount('/content/gdrive')
import os
# test是我网盘下的一个文件夹
os.chdir("/content/gdrive