首页 > 解决方案 > 从 Google Drive Colab 下载数据

问题描述

我一般是 TensorFlow 和 python 的初学者,所以任何帮助将不胜感激。我正在使用 tensorflow 中的本教程,仅使用我自己的数据。

所以我试图从我上传到谷歌驱动器的文件夹中获得的链接下载我自己的数据。然后,我将在图像分类器模型中使用该数据。但是,我开始看到图像被下载,它说:

dataset_url_training = "https://drive.google.com/drive/folders/genericid?usp=sharing" 
data_dir_training = tf.keras.utils.get_file('flower_photos', origin=dataset_url_training, untar=True)
data_dir_training = pathlib.Path(data_dir_training)

从https://drive.google.com/drive/folders/genericid?usp=sharing下载数据

106496/未知 - 2s 14us/步

然后它就停止了。当我尝试使用以下代码时:

print(image_count)

输出吐出:0

我真的很困惑,我不知道该怎么办。一些建议是制作一个 zip 文件 url,但这仅适用于单个文件,不适用于像我这样的整个文件夹。此外,据我所知,Google Drive 不允许您获取 zip 文件的链接,只能获取用于共享的链接(为了澄清,它们是我自己的文件)。

谢谢你。

编辑1:只是想清楚一点:我不是在寻找路径。我正在寻找一个 URL,因此使用了一个目录。我也尝试过使用。zip 文件的链接,但我收到与以前相同的错误消息。

标签: pythontensorflowkerasdownloadgoogle-colaboratory

解决方案


当您想在 colab 中使用 google drive 中的文件时,您可以将我们的驱动器挂载到 colab。

from google.colab import drive

drive.mount('/content/gdrive')

比你可以从谷歌驱动器打开文件。例如,您的文件位于主驱动器页面上的目录“文件夹”中:

path = "gdrive/My Drive/folder/flower_photos"

编辑/添加:

为了更清楚,您从教程中更改了这部分

import pathlib
dataset_url = "https://storage.googleapis.com/download.tensorflow.org/example_images/flower_photos.tgz"
data_dir = tf.keras.utils.get_file('flower_photos', origin=dataset_url, untar=True)
data_dir = pathlib.Path(data_dir)

对此

import pathlib
from google.colab import drive
drive.mount('/content/gdrive')
data_dir = "gdrive/My Drive/flower_photos"
data_dir = pathlib.Path(data_dir)

推荐阅读