我使用了library中的roi_pooling。
API格式为rpooling = roi_pooling(feature_map,rois,7,7)
。
就我而言,feature_map是(8,8,30,512)而rois是
[[ 0 15 2 27 2]
[ 1 10 5 22 6]
[ 2 18 5 27 6]
[ 3 13 4 24 5]
[ 4 3 2 8 3]
[ 4 2 3 10 3]
[ 5 18 5 26 5]
[ 6 8 2 18 2]
[ 7 16 1 24 2]]
由于第一列涉及要素地图索引,因此将其作为批处理中的索引。说第一列中的0表示批次中的第一张图像,而4表示批次中的第四张图像。由于第一列中有两个4,因此在第四个图像中显示两个rois。
我的考虑是否正确?