AlexeyAB darknet YOLOV3批量推理图片

最近使用AB版yolov3 推理一系列图片时无法进行。
引用自issue

1、
首先图片导出为txt,图片文件夹可以放在darknet工程下,也可在其他路径。txt内图片名称可以是绝对路径,也可以是工程下的相对路径。
2、
darknet/src/detector.c内 #include “option_list.h” 的下面添加

#include "memory.h

save_image(im, "predictions"); 下 且

 if (!dont_show) { 
     show_image(im, "predictions"); 
 } 

之前添加如下代码:`

char output[6];
memcpy(output, input + 4, 6);
char buff[256];
sprintf(buff, "results/%s", output);
save_image(im, buff);

memcpy函数详解

如果你的txt内图片地址为 fig/000001.jpg
则代码中output[6],memcpy(output, input + 4, 6);
4为从[fig/000001.jpg]的第四个索引开始复制6个字符
即000001
根据自己.txt中图片的路径改一下代码中的4,6

3、在darknet工程目录下打开命令行重新make

make

4、然后执行

./darknet detector test data/obj.data cfg/yolov3_2cls.cfg weights/yolov3_2cls_12000.weights -dont_show < filename.txt > result.txt

代码中filename.txt为你的.txt文件
会在result文件夹下看到图片。

你可能感兴趣的:(AlexeyAB darknet YOLOV3批量推理图片)