老卫带你学---faster-rcnn显示不出预测边框
在我们使用faster-rcnn的时候,会将预测边框画在图片上显示出来。
可是有的时候,我们并不能得到预测边框。
1.首先,我们需要检查我们demo.py的caffemodel与prototxt文件是否正确。
如图:
一定要是看清楚了!!!
2.如果上一步没有问题,我们就要考虑,是否因为图片本身的色彩空间稀薄导致提取特征过少。(因为老卫研究的是医学图象的计算机视觉,而医学照片是压缩的,相对而言与普通图像相比,空间显得单薄)
这个时候我们就要调整nms的阀值,让rpn尽可能可以预测更多的边框。
原始faster-rcnn中的阀值如下:
我们需要将其改小点,于是通过多次的实验,老卫将其改为如下,成功显示出结果。