darknet下利用resnet50訓(xùn)練貓狗大戰(zhàn)教程

https://blog.csdn.net/Obama_CSDN/article/details/81156590

1.darknet下對標(biāo)簽的要求

注意:路徑里不能包含標(biāo)簽名,否則會報Too many or too few labels錯誤

我一開始建立了目錄catdog,下面再建立了train和test目錄分別存放貓狗的文件,就報了上面的錯誤。我就把目錄名改成tacgod后,錯誤消失了。

舉個例子:park、parking-lot、road 和 roadside-parking-lot 存在標(biāo)簽重復(fù),必須保證標(biāo)簽字符串,互相不能被包含。可以修改為 apark、bparking-lot 和 croad,避免標(biāo)簽重復(fù)。


2. 修改cfg文件和data文件中的class數(shù)目為2

因為是二分類,所以要把最后一個濾波器改成數(shù)字2.


3.腳本

./darknet classifier train cfg/cat_dog.data cfg/resnet50_cat_train.cfg

./darknet classifier valid cfg/cat_dog.data cfg/resnet50_cat_test.cfg backup/resnet50_cat_train.weights

./darknet classifier predict cfg/cat_dog.data cfg/resnet50_cat_test.cfg backup/resnet50_cat_train.weights /home/mgl/darknet-master/tacgod/test/cat_1351.jpg

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容