Мне нравится внедрять ROIPooling в моей сети. После этой реализации в Keras.
В примере есть feature_maps_np = np.ones (feature_maps_tf.shape, dtype = 'float32') и roiss_tf = tf.placeholder (tf.float32, shape =(batch_size, n_rois, 4)). Заполнитель ROI находится в batch_size = 1, n_rois = 2 и 4 - это четыре угла.
Для моей сети batch_size равно 32. feature_maps_tf.shape равно [32,30,30,256]. roi_shape - это TensorShape ([Размер (32)]).
Но для некоторых изображений они представляют собой только одну область интереса, поэтому она имеет [230,350,275,360], но некоторые изображения имеют две области интереса и, например, [[230,350,275,360], [240,370,275,380]]
Как можно использовать ROIPoolingв партии?