darknet下利用resnet50训练猫狗大战教程

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

1.darknet下对标签的要求

注意：路径里不能包含标签名，否则会报Too many or too few labels错误

我一开始建立了目录catdog,下面再建立了train和test目录分别存放猫狗的文件，就报了上面的错误。我就把目录名改成tacgod后，错误消失了。

举个例子：park、parking-lot、road 和 roadside-parking-lot 存在标签重复，必须保证标签字符串，互相不能被包含。可以修改为 apark、bparking-lot 和 croad，避免标签重复。

2. 修改cfg文件和data文件中的class数目为2

因为是二分类，所以要把最后一个滤波器改成数字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