你能用鸽子注释文件夹中的所有图像吗答案

【问题标题】：can you use pigeon to annotate all images in a folder你能用鸽子注释文件夹中的所有图像吗
【发布时间】：2021-10-30 18:22:06
【问题描述】：

对于机器学习项目，我想用是或否标记文件夹中的图像。

我发现了这个很棒的工具；鸽子.. https://github.com/agermanidis/pigeon

但在线提供的示例要求用户提供脚本中所有文件的名称。我无法找到将脚本指向文件夹而不是包含所有图像名称的方法。

有没有办法只提供一个文件夹的链接。

from pigeon import annotate
from IPython.display import display, Image

annotations = annotate(
  ['assets/img_example1.jpg', 'assets/img_example2.jpg'],
  options=['cat', 'dog', 'horse'],
  display_fn=lambda filename: display(Image(filename))
)

【问题讨论】：

标签： python machine-learning computer-vision annotations data-annotations

【解决方案1】：

一种可能的解决方案是仅列出您感兴趣的文件夹中的所有文件。在这种情况下，您只需提供文件夹的路径（绝对或相对）并使用os.listdir 获取该文件夹中的所有文件。

然后我使用os.path.join 获取文件路径并将文件路径列表传递给annotate。

from pigeon import annotate
from IPython.display import display, Image
import os

path = "assets"
imgs = os.listdir(path)
imgs = [os.path.join(path, img) for img in imgs]

annotations = annotate(
  imgs,
  options=['cat', 'dog', 'horse'],
  display_fn=lambda filename: display(Image(filename))
)

【讨论】：

【解决方案2】：

我创建了一个遍历文件夹并使用图像名称填充列表的 for 循环。然后只是调用列表代替

['assets/img_example1.jpg', 'assets/img_example2.jpg']

【讨论】：