Faster RCNN代码记录(keras版)

参数:

num_rois: roi pooling层解析

# number of ROIs at once
# 训练classifer的回归框
# 经过rpn后输出的不止一个矩形框,所以这里我们是对多个ROI进行Pooling
"""
输出是batch个vector,其中batch的值等于RoI的个数,
vector的大小为channel * w * h;RoI Pooling的过程就是将一个个大小不同的box矩形框,
都映射成大小固定(w * h)的矩形框;
"""
self.num_rois = 4

猜你喜欢

转载自blog.csdn.net/weixin_31866177/article/details/83213698