在原YOLOv3仅识别一类目标的基础上，通过修改darknet/src/image.c中的draw_detections_v3函数，可以实现YOLOv4对多类目标的检测。在for循环中增加if条件，根据需要识别的类别数量嵌套相应数量的if判断，从而达到定制检测的目的。
博客中有人做了yolov3只检测人的代码：
yolov3 修改 只识别一类（person）
而自己需要其中两类检测结果，在上述博客作者的方法中做了一点小修改：
本人用的YOLOv4代码是在linux上运行的，这里贴上源代码地址：YOLOv4源码
需要改动的文件为 darknet/src 下的image.c文件
找到draw_detections_v3 函数，在第369行处：
qsort(selected_detections, selected_detections_num, sizeof(*selected_detections), compare_by_probs
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
最低0.47元/天 解锁文章
3325
