AceCoooool/RoIAlign-RoIPool-pytorch

Question about batch data.

kimna4 opened this issue · 1 comments

kimna4 commented

Hello,
First of all, really thank you for your hard work and sharing.

The reason why I write the issue is that I don't know exactly a meaning of "batch index" (the first element of "rois")

I hope to use your roi_align (or roi_pooling) to batch tensor. In that case, how to use your function?

Thank you

AceCoooool commented
  1. the batch_index means: which batch does this roi belong to. (for example, if there only one "object", the batch_index is always 0)
  2. you can reference simple-faster-rcnn-pytorch, faster-rcnn.pytorch to know how to use this function.