1. 上传数据
将数据上传到Google云端硬盘
2. 代码中挂载云端硬盘
挂载驱动器中的数据,需要在Colab的文件中使用以下代码。以python文件为例:
from google.colab import drive
drive.mount('/content/drive')
此时,会在下方出现验证码输入框。复制链接到浏览器,授权后即可查找到验证码,输入即可。
这样就可以将云端硬盘Drive挂载到Colab
3.读取对应目录的数据集
image_datasets = {x: datasets.ImageFolder("./content/drive/MyDrive/datasets/XX", data_transforms[x]) for x in ['train', 'val']}
这样就大功告成了。整个过程类似于linux中加载移动硬盘的过程。