手动标记图像

Question

我以这种方式创建了一个图像列表作为np.array：Dataset =（1000，100，100，3），在这种情况下，我有1000个3d数组，每个数组代表一个100x100 RGB图像。

我要做的是扫描每张图像并为其分配一个标签，例如0,1,2

到目前为止，我尝试过类似的操作：

labels = np.zeros( 1000, 1) )

fig, ax = plt.subplots()
for k in range(0, 1000):
    #show image
    plt.imshow(Dataset[k,:,:,:])
    plt.show()
    plt.pause(1)
    # assign label
    labels[k] = input("Which image is it? 0, 1, 2?)

但是，如图所示，该过程确实很麻烦，但是，一旦我在控制台上单击以键入标签，图像就会消失为图标，然后在下一次迭代中仍然保持隐藏状态。

我认为有更好的方法可以做到这一点...