使用Google Colab时如何从Google drive中加载自定义的包模型和数据集

Posted 2020-12-19 flyingtester

在Google colab上进行推理时，需要把数据和训练好的模型放到google drive中，然后进行加载。

首先，需要在google drive中建一个目录存放模型，在这里，我选择默认的文件夹Colab Notebooks，为了方便起见，最好把模型和需要用到的自定义包压缩到一起上传

 

上传之后新建一个colab notebook

首先需要加载google drive，用如下代码

1 import os
2 from google.colab import drive
3 drive.mount(‘/content/drive‘)

运行之后会出现一个让你授权的链接，右键新窗口打开，点击allow之后将authorization code复制到notebook的输入框中并回车

 

使用如下命令进入到刚才上传模型的文件夹，此处Colab Notebooks替换成你实际上传的文件夹

!cd /content/drive/My Drive/Colab Notebooks/

!ls

由于要在colab中执行linux命令，所以需要在前面加个叹号

 使用unzip命令来解压刚才上传的压缩包

!unzip XXXX.zip

此处将XXXX替换成你的压缩文件名称

解压完成后可以看到在google drive中也多了对应的文件夹

如果需要加载自定义的包，刚在notebook中加上如下两句，其中ThirdPartyModule是我用来存放第三方自定义包的目录，此处可以替换成你自己的，其中

models是我放的自定义包的名称

import sys
sys.path.append(‘/content/drive/My Drive/Colab Notebooks/ThirdPartyModule‘)

import models

如果做了前面的操作，加载模型和数据时，只需写好对应的路径即可，如

checkpoint = torch.load(‘/content/drive/My Drive/Colab Notebooks/XXX/pre_trained.tar‘, map_location=torch.device(‘gpu‘))

和

b = cv2.imread(‘/content/drive/My Drive/Pics/2020.jpg‘)

接下来，Just enjoy your trip of Google Colab!