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

标签：路径   自己的   location   出现   check   右键   style   mode   方便   

在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替换成你实际上传的文件夹

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

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‘)

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

标签：路径   自己的   location   出现   check   右键   style   mode   方便   

原文地址：https://www.cnblogs.com/flyingtester/p/12879810.html