修改YOLO使其显示自定义类别

基本参考自这篇文章（http://blog.youkuaiyun.com/ma3252788/article/details/74659230），主要用来记录下自己遇到的问题

 

根据@赤战约风 的帖子做如下修改可以使其只显示人这个类别

一.修改源代码

cfg/coco.data

classes= 1     #修改成1
train  = /home/pjreddie/data/coco/trainvalno5k.txt
valid  = coco_testdev
#valid = data/coco_val_5k.list
names = data/coco.names backup = /home/pjreddie/backup/ eval=coco

 

src/detector.c

void run_detector(int argc