FCN的分類的label是直接寫類號碼：1,2,3,...,255。
但softmaxwithLoss的第一個bottom，也就是網絡輸出，是要求向量。
即label=3的話，輸出是0,0,0,1,0,0,...
所以寫腳本注意一下，輸出層的輸出個數不能寫1。
就是說softmaxwithLoss的兩個bottom第二維數值是不一樣的。
caffe下FCN注意事項
發表評論
所有評論
還沒有人評論，想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章
FCN的分類的label是直接寫類號碼：1,2,3,...,255。
但softmaxwithLoss的第一個bottom，也就是網絡輸出，是要求向量。
即label=3的話，輸出是0,0,0,1,0,0,...
所以寫腳本注意一下，輸出層的輸出個數不能寫1。
就是說softmaxwithLoss的兩個bottom第二維數值是不一樣的。