我研究了与功能图的RoI池相关的各种问题,但是我无法在Keras中正确地编写出来。
我使用了this repository中提供的代码,但是遇到了一些兼容性问题。
我正在从预先训练的VGG 16模型中提取特征图,并且如Faster R-CNN论文所述,我从RPN层获得了我的RoI边界框,但是我无法对RoI池化层进行编码。
任何人都可以提供工作代码或源代码来进行相同的工作吗?
Tensorflow实现也受到高度赞赏。如果需要其他信息,请发表评论。
供参考的代码如下:
model = VGG16()
feature_map=model.predict(img)
feature_map = np.asarray(feature_map, dtype='float32')
rois_value = [
[0, 0, 0, 1, 3],
[0, 2, 2, 3, 3],
[0, 1, 0, 3, 2]
]
rois= np.asarray(rois_value, dtype='int32')
rpooling = roi_pooling(feature_map, rois, 7, 7)
#roi_pooling will be the function that takes feature-map, the roi bounding box coordinates(the later four being startx, starty, width, height) and pooling height and pooling width.