Google Colab：如何循环浏览文件夹中的图像？

Question

我已经将大量 .jpg 图像从我的 Google Drive 复制到 Google Colab per Google Colab: how to read data from my google drive? :

 local_download_path = os.path.expanduser('~/data')
try:
  os.makedirs(local_download_path)
except: pass

# 2. Auto-iterate using the query syntax
#    https://developers.google.com/drive/v2/web/search-parameters
file_list = drive.ListFile(
    {'q': "'1SooKSw8M4ACbznKjnNrYvJ5wxuqJ-YCk' in parents"}).GetList()

for f in file_list:
  # 3. Create & download by id.
  print('title: %s, id: %s' % (f['title'], f['id']))
  fname = os.path.join(local_download_path, f['title'])
  print('downloading to {}'.format(fname))
  f_ = drive.CreateFile({'id': f['id']})
  f_.GetContentFile(fname)

这已完成，因此所有图像文件都应位于 data 文件夹中。我现在如何实际访问这些文件？我想遍历 data 文件夹中所有类型为 .jpg 的文件。

原文由 ZhouW 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

将数据下载到 local_download_path ，您可以使用 glob.glob 通过扩展名获取文件：

 import glob
images = glob.glob(local_download_path + '/*.jpg')

原文由 taras 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 2

一种替代方法是使用已经实现的 os 库。

 # Just needed in case you'd like to append it to an array
data = []

for filename in os.listdir(local_download_path):
    if filename.endswith("jpg"):
        # Your code comes here such as
        print(filename)
        data.append(filename)

原文由 RandomDude90 发布，翻译遵循 CC BY-SA 4.0 许可协议