我有一個數據集,其中圖像有變化標籤數量。標籤的數量在1到5之間。共有100個類別。Caffe:具有不同標籤數量的多標籤圖像
谷歌搜索後,看起來像HDF5分貝層分貝可以處理多個標籤,如下面的URL。
唯一的問題是它假設有一個固定數量的標籤。在此之後,我將不得不創造非標類一1×100矩陣,其中,項值爲1的標記類,0,如以下定義:
layers {
name: "slice0"
type: SLICE
bottom: "label"
top: "label_matrix"
slice_param {
slice_dim: 1
slice_point: 100
}
}
其中每個圖像包含AA標籤找像(1,0,0,... 1,... 0,...,0,1),其中矢量大小爲100維。
現在,我很抱歉,我的問題變得模糊,但這是一個可行的想法?也就是說,有更好的方法來解決這個問題嗎?
您是否建議每張圖片的標籤可寫爲[5,10,25,32,91]?關於PO的問題,我只知道將第5,10,25,32和91st的標籤向量寫爲100-dim向量的等價方法,其值爲'1',其餘爲'0'。你可以提供任何參考嗎?謝謝! – mintaka
對不起,現在不認爲我有參考。您可以將標籤向量定義爲長二進制代碼,並使用sigmoidcrossentropy丟失圖層。這將允許標籤矢量中的兩個類同時打開,或者爲每個「類型」的標籤定義一個損失層。這樣你可以保持標量標籤的值,而不必將它們轉換成二進制向量並將它們連接起來。 – ypx