使用Python根据用户输入标记图像

Question

我有1000张属于类别1或2的图像。我想对其进行迭代并手动标记它们。

例如，一旦我开始运行代码，弹出第一张图像，我按“ Enter”进入类别1或按“ Esc”进入类别2，然后该图像将自动关闭，下一张图像将弹出...此过程不断重复，直到到达最后一张图片为止。

有人对我可以使用哪个包装有任何建议吗？目前，我只找到tkinter。谢谢！

Answer 1

您也可以将OpenCV用于此目的。

import cv2

for img in images:
    cv2.imshow("Please select category", img)

    # wait until key pressed
    key = cv2.waitKey()

    # esc key
    if key == 27:
        cat = 2
    else:
        cat = 1

# close all the windows
cv2.destroyAllWindows()