老卫带你学---faster-rcnn显示不出预测边框

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/yixieling4397/article/details/81606296

在我们使用faster-rcnn的时候,会将预测边框画在图片上显示出来。
可是有的时候,我们并不能得到预测边框。

1.首先,我们需要检查我们demo.py的caffemodel与prototxt文件是否正确。
如图:
这里写图片描述

这里写图片描述

一定要是看清楚了!!!

2.如果上一步没有问题,我们就要考虑,是否因为图片本身的色彩空间稀薄导致提取特征过少。(因为老卫研究的是医学图象的计算机视觉,而医学照片是压缩的,相对而言与普通图像相比,空间显得单薄)

这个时候我们就要调整nms的阀值,让rpn尽可能可以预测更多的边框。

原始faster-rcnn中的阀值如下:

这里写图片描述

我们需要将其改小点,于是通过多次的实验,老卫将其改为如下,成功显示出结果。

这里写图片描述

这里写图片描述

猜你喜欢

转载自blog.csdn.net/yixieling4397/article/details/81606296