search

longcw / RoIAlign.pytorch

RoIAlign & crop_and_resize for PyTorch
555 stars 103 forks source link

It supports batchsize>1？ #16

Open gjd2017 opened 6 years ago

gjd2017 commented 6 years ago

Hi, dose it support batchsize>1?

longcw commented 6 years ago

Yes, it supports batch > 1. You can check the example here: https://github.com/longcw/RoIAlign.pytorch/blob/master/tests/test.py#L114