前言
为表示尊敬,先贴出来两位大神的源码地址:
YOLOV3keras实现地址:https://github.com/qqwweee/keras-yolo3
MAP计算:https://github.com/Cartucho/mAP
本篇文章用到的数据是博主之前的文章《【YOLO初探】之 keras-yolov3训练自己数据集》的数据,测试也用的是那篇文章训练好的权重文件。用了230图片做了个数据集,20张用来测试。效果非常好,感叹YOLO大法好!
本文的内容主要是如何计算MAP,什么是MAP,请查看博主的《目标检测问题中的-MAP》,这里不再赘述!
测试下作者数据
下载好https://github.com/Cartucho/mAP文件以后,解压,运行 main.py,接下来就是“劈里啪啦”见证奇迹的时刻了:
运行结果保存在了当前目录下的'./results'文件夹中
======================= RESTART: G:\mAP-master\main.py =======================
22.73% = backpack AP
85.94% = bed AP
17.52% = book AP
14.29% = bookcase AP
23.48% = bottle AP
31.86% = bowl AP
7.93% = cabinetry AP
53.84% = chair AP
4.55% = coffeetable AP
19.05% = countertop AP
42.50% = cup AP
39.66% = diningtable AP
0.00% = doll AP
20.69% = door AP
7.69% = heater AP
71.43% = nightstand AP
42.86% = person AP
17.71% = pictureframe AP
13.01% = pillow AP
62.31% = pottedplant AP
73.21% = remote AP
0.00% = shelf AP
16.33% = sink AP
90.48% = sofa AP
1.39% = tap AP
0.00% = tincan AP
63.25% = tvmonitor AP
18.75% = vase AP
45.45% = wastecontainer AP
23.53% = windowblind AP
mAP = 31.05%