tensorflow - 如何使用已安装的 Google 驱动器中的 Tensorflow 在 Google Colab 笔记本中加载图像

Question

在 Google Colab 笔记本中，我安装了我的 Google 驱动器并且可以看到我的文件。我正在尝试加载一个压缩目录，该目录有两个文件夹，每个文件夹中有几个图片文件。

我遵循了 Tensorflow 网站上的一个示例，其中有一个关于如何加载图片的示例，但它使用的是远程位置。这是网站 - https://www.tensorflow.org/tutorials/load_data/images

下面是示例中有效的代码:

data_root_orig = tf.keras.utils.get_file(origin='https://storage.googleapis.com/download.tensorflow.org/example_images/flower_photos.tgz',
                                        fname='flower_photos', untar=True)
data_root = pathlib.Path(data_root_orig)
print(data_root)

这是修改后的代码，我在其中尝试从已安装的 Google 驱动器中引用压缩目录:

data_root_orig = tf.keras.utils.get_file(origin='/content/gdrive/My Drive/TrainingPictures/',
                                         fname='TrainingPictures_Car', untar=True)
data_root = pathlib.Path(data_root_orig)
print(data_root)

我收到这个错误:

ValueError: unknown url type: '/content/gdrive/My Drive/TrainingPictures/'

它显然需要一个 URL 而不是我提供的路径。我想知道如何加载 Google 驱动器提供的压缩目录。

Answer 1

在这种情况下，不需要使用tf.keras.utils.get_file()，只需要Path即可。

这里有两种方法可以做到这一点

首先:!unzip -q 'content/gdrive/My Drive/TrainingPictures/TrainingPictures_Car.zip'

解压到'/content/'

import pathlib
data = pathlib.Path('/content/folders_inside_zip') 
count = len(list(data.glob('*/*.jpg')))
count

第二个:如果存档已经在谷歌驱动器中解压缩:

import pathlib
data = pathlib.Path('/content/gdrive/My Drive/TrainingPictures/') 
count = len(list(data.glob('*.jpg')))
count