【YOLO】只检测人

一.修改源代码

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

 原因：在coco.c中定义的标签，人为第一个，所以改为1.

examples/detector.c

void test_detector(char *datacfg, char *cfgfile, char *weightfile, char *filename, float thresh)函数中 draw_detections(im, l.w*l.h*l.n, thresh, boxes, probs, names, alphabet, 1); //最后一个参数修改成1。line=481 void run_detector(int argc, char **argv)//该函数中，倒数第三行.line=542 int classes = option_find_int(options, "classes", 1);//最后一个参数修改成1  

二.重新编译和测试

1.重新编译

make clean make

2.测试

./darknet detect cfg/yolo.cfg yolo.weights data/person.jpg

三.结果

 

转载于:https://www.cnblogs.com/tenderwx/p/7278447.html

相关资源：YOLOv3检测行人结果（caltech reasonble）