1. 程式人生 > >CNN影象分類中類別數量確定

CNN影象分類中類別數量確定

影象分類中最終將影象分為多少類,一般是由CNN結構中最後一個全連線層(FC)神經元個數確定,即要將影象分為多少類最後一個全連線層就為多大,例如VGG、ResNet等,與最後的softmax無關。
LeNet網路中最後沒有用到softmax,而是將最後一個FC的結果作為輸入,通過歐式徑向基函式(Euclidean Radial Basis Function)單元進行計算得到最後的分類結果,其中歐式徑向基函式(Euclidean Radial Basis Function)單元的個數就是類別數量。