you359/Keras-FasterRCNN

RoiPooling with batch_size > 1

AlbertoSabater opened this issue · 1 comments

Hi! According to with the comments, the RoiPooling layer works with input size (1, rows, cols, channels), does that mean that it only works with batch_size = 1, or can it be higher ((batch_size, rows, cols, channels))?
Thanks in advance

set it for(-1, self.num_rois, self.pool_size, self.pool_size, self.channels), right?I training to add mult-batch, but something wrong.