RoIAlign-RoIPool-pytorch
RoIAlign-RoIPool-pytorch copied to clipboard
Question about batch data.
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
- the batch_index means: which batch does this roi belong to. (for example, if there only one "object", the batch_index is always 0)
- you can reference simple-faster-rcnn-pytorch, faster-rcnn.pytorch to know how to use this function.