RoIAlign-RoIPool-pytorch icon indicating copy to clipboard operation
RoIAlign-RoIPool-pytorch copied to clipboard

Question about batch data.

Open kimna4 opened this issue 7 years ago • 1 comments

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

kimna4 avatar Aug 30 '18 11:08 kimna4

  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.

AceCoooool avatar Aug 30 '18 12:08 AceCoooool