使用Python根据用户输入为图像添加标签

我有1000张属于第1类或第2类的图片。我想迭代它们并手动标记它们

例如，一旦我开始运行代码，第一个图像就会弹出，我按“回车”键它属于类别1，或按“Esc”键它属于类别2。然后图像将自动关闭，下一个图像将弹出…此过程将不断重复，直到到达最后一个图像

---

有人对我可以使用哪种包装有什么建议吗？目前，我只找到tkinter。谢谢

您也可以使用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()

---

特金特怎么了？看起来它应该满足你的要求。太好了！我以前从未用过tkinter。我想确定它是我应该使用的正确的软件包。