classifier[6].out_features
時間：2022-03-16 16:15:01 | 來源：行業(yè)動態(tài)
時間：2022-03-16 16:15:01 來源：行業(yè)動態(tài)
是模型最終的輸出數(shù)量，也就是分類數(shù)量。
- 因為AlexNet這個深度神經(jīng)網(wǎng)絡(luò)的鼻祖，是在2012年ILSVRC競賽中以1000分類的ImageNet數(shù)據(jù)集作為測試標(biāo)，一舉拔得頭籌而開創(chuàng)深度學(xué)習(xí)新時代，這個1000分類的AlexNet圖像分類模型便成為這個領(lǐng)域的經(jīng)典之作，因此在標(biāo)準(zhǔn)的AlexNet模型中就保留1000這個數(shù)字作為基準(zhǔn)。
- 在這個避障應(yīng)用中只使用free與blockerd兩個分類，因此需將classifier[6]的輸出類別數(shù)量調(diào)整為2。請在代碼下方加入print(model.classifier[6])指令，打印出修改后的內(nèi)容，會看到out_features的值已經(jīng)變成2。
在線咨詢