2017-02-17 47 views
2

我想implememt多任務神經網絡在tensorflow,爲此,我需要輸入:如何在張量流中給圖像添加多個標籤?

[圖像label1的標籤2]

,我可以給你的訓練神經網絡。

我的問題是,如何將TFRecord文件中的多個標籤與圖像關聯?

我目前使用的是build_image_data.py用於生成TFRecord文件的初始模型文件,但在這種情況下,每個圖像只有一個標籤。

回答

2

他們總是有兩個標籤嗎?如果是這樣,請將「label1-label2」作爲一個標籤。 或者乾脆建立兩個網絡,一個用於標籤1,另一個用於標籤2.

它們是分層標籤嗎?然後,檢查分層分類器。

0
  1. 我得到了這個工作。對於查找引用的任何人,您可以修改build_image_data.py文件的Example proto並將其與兩個標籤相關聯。 :)
相關問題