本文共 636 字，大约阅读时间需要 2 分钟。

YOLO-V2实现识别单类

源码修改

修改cfg/voc.data

classess=20 改成 classes = 1

修改data/voc.names

只留下person这一类

修改examples/detector.c

1. 修改run_detector函数

void run_detector(int argc, char **argv)//该函数中，第line=849行int classes = option_find_int(options, "classes", 1);//最后一个参数修改成1

void test_detector(char *datacfg, char *cfgfile, char *weightfile, char *filename, float thresh, float hier_thresh, char *outfile, int fullscreen)//函数中，第line=604行draw_detections(im, dets, nboxes, thresh, names, alphabet, l.classes); //l.classes改成 1

重新编译和测试

1. 编译

make cleanmake

1. 测试

./darknet detect cfg/yolov3.cfg  cfg/yolov3.weights /home/it/Yolo/crop001007.png

参考文献