2016-03-15 34 views
9

TensorBoard是一個很好的工具,但它可以更強大嗎?下圖顯示了TensorBoard中的可視化。如何在TensorBoard圖像上添加標籤?

它是由下面的代碼稱爲:

tf.image_summary('images', images, max_images=100) 

作爲API所暗示的,最後一位數字是「圖像數」,因爲我指定max_images,從0到99在這種情況下= 100。我想問,如果我可以將這個圖像的標籤附加到文本中?這將是一個很棒的功能,因爲它允許用戶在訓練過程中實時看到圖像及其各自的標籤。在某些圖像完全貼錯標籤的情況下,可以執行修復。換句話說,我想在下面的圖片中的相應文字是:

images/image/9/5 
images/image/39/6 
images/image/31/0 
images/image/30/2 
where last digit is the label. 

謝謝!

enter image description here

回答

6

我一直沒能找到一種方法來做到這一點只用tensorflow,所以不是我做了以下內容:

  1. 創建摘要圖像佔位符(例如像(10,224,224,3)用於十個概要圖像)。
  2. 根據該佔位符創建圖像摘要。
  3. 在驗證過程(或者培訓過程中,如果你喜歡的話),使用類似session.run([sample_images, sample_labels])的東西,將圖像和摘要標籤拉成python。
  4. 迭代遍歷批次並使用OpenCV將標籤寫入圖像使用cv2.putText
  5. 運行提供佔位符標籤圖像的摘要操作。
+1

雖然這可能是一個很好的黑客大圖像,但它可能不適用於(28,28)圖像。 – jkschin

+4

沒有理由總結必須是28x28。你可以使用任何你喜歡的尺寸。您可以從128x28的空白畫布開始,將訓練示例複製到左上角,並使用剩餘的100x28矩形繪製文本。 –

+0

那麼我必須說這是一個合法的黑客呢! – jkschin

相關問題