你能使用pigeon为文件夹中的所有图像添加注释吗

Question

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

我发现了一个很棒的工具: Pigeon..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

一种可能的解决方案是只列出您感兴趣的文件夹中的所有文件。在本例中，您只需提供文件夹的路径(绝对路径或相对路径)，并使用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))
)

Answer 2

我创建了一个for循环，它遍历了一个文件夹，并用图像名称填充了一个列表。然后调用list来代替

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