在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!