采用的代码是YOLOV4官方源码,地址为:https://github.com/AlexeyAB/darknet
论文链接:https://arxiv.org/abs/2004.10934
具体的配置和训练到检测可以参考我的上一篇博客:YOLOV4训练自己的数据集—从环境配置到完成检测任务(Windows)
这篇主要是针对在批量测试时,我们可能有一种需求就是:将批量检测完成后的图片保存到自己的文件夹中,同时检测的标签以文本文件的方式存储。(也可以看这个代码)
首先在源码文件夹中找到src/detector.c源文件 将detector.c的test_detector函数整体替换,然后重新编译darknet,生成新的darknet.exe文件。控制台测试命令:darknet.exe detector test data/obj.data cfg/yolo-obj.cfg backup/yolo-obj_final.weights data/test.txt