【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

三.結果