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

最新推荐文章于 2025-07-25 20:15:34 发布

原创最新推荐文章于 2025-07-25 20:15:34 发布 · 3.5k 阅读

5 ·

CC 4.0 BY-SA版权

文章标签：

#目标检测

深度学习同时被 2 个专栏收录

17 篇文章

订阅专栏

YOLO相关，目标检测

1 篇文章

订阅专栏

本文介绍如何通过修改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

src/detector.c

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

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

二.重新编译和测试

1.重新编译

make clean
make

2.测试

./darknet detector test cfg/coco.data cfg/yolo.cfg yolo.weights data/dog.jpg

这样的修改其实还是有点麻烦的，参考yangshun8889大神的回答，其实还可以这样：

我假设你使用的是VOC.name, 在VOC中，car对应着6, person对应着14,然后修改代码。在draw_detections()函数中第一个for循环int class = max_index(probs[i], classes);代码后面添加if(class!=6 && class!=14){continue;}

这样就可以检测人和车了。

参考链接：