你能用Pigeon来注释文件夹中的所有图像吗

Question

我想为机器学习项目的文件夹中的图像添加“是”或“否”标签

我发现了这个伟大的工具；鸽子 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))
)

Answer 1

一种可能的解决方案是只列出您感兴趣的文件夹中的所有文件。在本例中，只提供文件夹的路径（绝对或相对），并使用^{}获取该文件夹中的所有文件

然后，我使用^{}获取文件路径，并将文件路径列表传递给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))
)

Answer 2

我做了一个for循环，遍历了一个文件夹，并用图像名填充了一个列表。然后就用名单代替了

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