如何使用Pigeon标注工具无需逐个输入文件名即可标注文件夹内所有图片

Pigeon本身未内置直接传入文件夹路径的参数，但你可以通过Python内置的os模块快速读取目标文件夹下所有图片路径，生成路径列表后直接传入annotate接口即可，无需手动录入文件名。

完整可运行代码 #

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

# 配置参数：替换为你的待标注文件夹路径
IMAGE_FOLDER = "./your_image_folder"
# 支持的图片后缀，可按需增减
SUPPORT_EXTS = ('.jpg', '.jpeg', '.png', '.bmp')

# 自动遍历文件夹生成图片路径列表
image_paths = [
    os.path.join(IMAGE_FOLDER, filename) 
    for filename in os.listdir(IMAGE_FOLDER)
    if filename.lower().endswith(SUPPORT_EXTS)
]

# 调用标注接口，标签按需求设置为yes/no
annotations = annotate(
  image_paths,
  options=['yes', 'no'],
  display_fn=lambda filename: display(Image(filename))
)

补充说明 #

• 上述代码默认不会递归遍历子文件夹，如果需要读取子文件夹内的图片，可以把列表生成逻辑替换为os.walk实现
• 标注完成后返回的annotations是由（图片路径，标签）组成的列表，你可以自行导出为csv或json格式做后续存储使用

内容的提问来源于stack exchange，提问作者unaied