0. 写作目的
好记性不如烂笔头。
1. 有关matplot 绘制曲线图
这里以绘制目标检测中P-R(Precision - Recall)曲线为例进行说明。
具体如何计算目标检测中的AP,和mAP请参考我的另一博客。
下面的代码是SSD/caffe版本中的caffe/exampels/ssd/score_ssd_pascal.py中的增加的绘制P-R曲线的片段。且保存了输出过程中最大Precsion。(这里绘制的P-R不是原始意义上的P-R曲线,绘制的是recall(0.0 0.1 ... 0.9 1.0)和最大precision的P-R曲线,由于计算AP就是采用此计算方法,因此可以近似看成P-R曲线)。
以下代码大体思路为:先从caffe输出的log中,找出输出最大MaxPrecision的值(此处需要修改caffe源码来便于找到带有MaxPrecision值得日志行),然后将这些MaxPrecision保存到txt文件中,便于后续使用(由于代码中给出的绘图大小为10*10,如果太小或者太大,都可以依据txt文件重新绘制)。最后对于不同的类,绘制不同的P-R曲线,并保存。<