Warm tip: This article is reproduced from serverfault.com, please click

python-更快的RCNN锚生成

(python - Faster RCNN Anchor Generation)

发布于 2020-12-05 08:27:31

我想理解Faster RCNN的概念。

例如,在一幅图像(224×224)中,只有两个对象。要从图像创建长度为256(128前景,128背景)的锚的迷你批处理,与地面真值边界框相比,我只得到30个IOU大于0.7的锚。

在这种情况下,如何使前景对象与背景保持平衡?

Questioner
user12914245
Viewed
0
David S 2020-12-05 16:52:16

你可以摆脱或设置前景与背景之间的预定义比例。

在下面的链接中,他将前景与背景的比例设置为1/3。

https://dongjk.github.io/code/object+detection/keras/2018/05/21/Faster_R-CNN_step_by_step,_Part_I.html

本教程的github是:

https://github.com/dongjk/faster_rcnn_keras/blob/master/RPN.py

这是一个完整的教程,它在训练Faster-RCNN之前逐步完成了所有步骤,在你的情况下,GitHub中的RPN脚本提供了你所需要的解决方案。

请注意,你并不需要完全平衡,而是要有一个合理的比率,因为在非常特定的数据集的情况下,大多数情况下背景是图像的主要部分,而事实并非如此。