Tensorflow Inception v3重新培訓-將文本/標簽附加到單個圖像

Question

我正在使用Inception v3模型重新訓練自己的數據集。 我的文件夾很少，代表每個班級都有圖像的班級。 我想做的是將一些文本ID“附加”到這些圖像上，這樣當它們經過重新訓練並用於運行分類/相似性檢測時，這些ID也會被檢索到。 （基本上是圖像相似度檢測）

例如，圖像X屬於“老師”類，它屬於John。 當我重新訓練模型並在新模型上進行分類時，我希望獲得教師班級，但除此之外，我還想知道誰是老師（約翰）。

任何想法如何去做？

問候

Answer 1

您有3個主要選擇-增加課程，多標簽學習或訓練幾種模型。

第一種選擇是最直接的-可以讓班級為Teachers_John的班級和班級為Teachers_John的班級而不是擁有屬於John的老師和屬於Jane的老師，並像學習其他任何課程一樣學習分類到這些類別類別集，或使用類似等級的softmax。

第二種選擇是設置一組類別，其中包括教師以及約翰和簡-現在，您的目標不是正確預測一個最准確的班級（教師），而是幾個（老師和約翰）。

您的最后一個選擇是創建模型層次結構，其中第一個模型學習區分John和Jane以及其他模型，以對每個模型的內部類進行分類。