本篇为yolov3源码配置实现部分及 使用官方模型进行批量测试
目录
文章目录
源码配置
- 下载源码并编译
git clone https://github.com/pjreddie/darknet
cd darknet
make
- 下载预训练权重(coco)
wget https://pjreddie.com/media/files/yolov3.weights
- 单张图片测试
darknet下
./darknet detect cfg/yolov3.cfg yolov3.weights data/dog.jpg
应该会出现这个界面:
再然后会保存到darknet下prediction.jpg,不会显示,因为没有在opencv下编译
好的,这是单次测试,然后是批量测试
批量测试
如果只是想要批量测试一下并输出记录目标位置信息的txt文件,可以使用valid命令,如下:
./darknet detector valid cfg/voc.data cfg/yolov3.cfg yolov3.weights
其中,cfg/voc.data 为测试训练路径等配置,这里需要修改
而cfg/yolov3.cfg为yolov3网络配置文件
yolov3.weights 为刚刚下载的权重文件
classes= 20
train = /home/pjreddie/data/voc/train.txt
valid = /home/dbc/darknet/test.txt #你自己的测试图像路径
names = data/coco.names #注意这里使用的是coco所训练的模型yolov3.cfg所以这里对应为coco.names
backup = backup
这里test.txt为测试图像的路径,
可以使用命令
ls -R /home/dbc/DATASET/1/pic/*.jpg > files.txt
将路径下文件的绝对路径写入files.txt文件中。
我的test.txt中内容如下:
/home/dbc/DATASET/1/pic/000001.jpg
/home/dbc/DATASET/1/pic/000002.jpg
/home/dbc/DATASET/1/pic/000003.jpg
/home/dbc/DATASET/1/pic/000004.jpg
/home/dbc/DATASET/1/pic/000005.jpg
/home/dbc/DATASET/1/pic/000006.jpg
/home/dbc/DATASET/1/pic/000007.jpg
/home/dbc/DATASET/1/pic/000008.jpg
……
执行完在./results/comp4_det_test_[类名].txt里会保存测试结果,像这样
打开comp4_det_test_[handbag].txt看一下:
按列,分别为:图像名称 | 置信度 | xmin,ymin,xmax,ymax
如果想要批量测试并保存测试结果则需要修改./example/detector.c文件
这里参考了博客
https://blog.csdn.net/mieleizhi0522/article/details/79989754
中的修改方法:
- 使用以下函数替换了detector.c中的void test_detector函数, 注意修改3处路径
void test_detector(char *datacfg, char *cfgfile, char *weightfile, char *filename, float thresh, float hier_thresh, char *outfile, int fullscreen)
{
list *options = read_data_cfg(datacfg);
char *name_list = option_find_str(options, "names", "data/names.list");
char **names = get_labels(name_list);
image **alphabet = load_alphabet();
network *net = load_network(cfgfile, weightfile, 0);
set_batch_network(net, 1);
srand(2222222);
double time;
char buff[256];
char *input = buff;
float nms=.45;
int i=0;
while(1){
if(filename){
strncpy(input, filename, 256);
image im = load_image_color(input,0,0);
image sized