记录使用AlexeyAB大神darknet下的ap计算(Ubuntu)
使用reval_voc_py3.py计算AP值
1、文件位置
两处位置上有reval_voc_py3.py和voc_eval_py3.py(也有python版,分别是reval_voc.py和voc_eval.py)
(1)./darknet/scripts(所使用的)
(2)./darknet/build/x64(这里只有python3版本的)
2、运行darknet中的detector valid指令,生成对测试集的检测结果。
./darknet detector valid <data文件路径> <cfg文件路径> <weights文件路径>
运行完后会在darknet项目下的results生成一个comp4_det_test_<自己类别>.txt的检测结果文件
3、所使用的命令及参数
python3 reval_voc_py3.py --voc_dir <voc文件路径> --year <年份> --image_set <验证集文件名> --classes <names文件路径> ./results
我在–voc_dir中使用的是自己单独出来的测试目录。所以在测试目录里面需要添加文件夹VOC2017/Annotations,该文件夹存放测试图像的xml文件;VOC2017/ImageSets/Main,该文件存放测试图片的文件名,形如以下(文件名不含后缀):