如何顯示錯誤分類的圖像？

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)