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