tensorflow - 如何显示错误分类的图像？

Question

在 TensorFlow 的一般训练管道中，如何打印(可能通过 TensorBoard)错误分类的示例/图像？

Answer 1

假设您正在使用一个 softmax 分类器，它在 N 个类之间进行选择作为网络的最后一层。伪代码可能看起来像这样，其中最后一层的第一个维度是批量大小:

# computation graph
predictions = argmax(softmax(final_layer))
matches = predictions == argmax(labels) # if one-hot encoded

# later
batch_matches = sess.run(matches, feed_dict={...})

for image, does_match in zip(batch_images, batch_matches):
  if not does_match:
    cv2.imwrite('mismatched.png', image)