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