如何将一个仅包含图像的 Google 云端硬盘文件夹挂载到 Google Colab 并遍历它？

Question

我有一个为计算机视觉项目提取的图像文件夹，但不知道如何将其装载到 Google Colab 上并遍历该文件夹。 我有一个功能需要应用于文件夹内的每个图像，但不知道如何获取图像。

我曾尝试寻找资源，但没有找到对我的情况有帮助的任何资源，因为它们中的大多数都是用于解压缩不是图像的文件。 你能帮帮我吗？ 谢谢你。

Answer 1

您可以为此使用 OpenCV 库。

from google.colab import drive
import os
import cv2

首先，您需要将当前工作目录更改为图像文件夹目录。

os.chdir("/content/drive/MyDrive/yourfolder")

您可以遍历每个图像，将您的函数应用于它们并保存最终版本，如下所示：

for file in os.listdir():
  img = cv2.imread(file)
  
  result = myfunction(img)
  
  cv2.imwrite(file, result)