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

©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容