与图像特征图上的RoI池相关的问题

时间:2019-06-30 17:14:46

标签: python keras neural-network conv-neural-network roi

我研究了与功能图的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.

0 个答案:

没有答案