Rotated_ROIAlign
Rotated RoIAlign in pytorch, implemented with CUDA. This implementation is edited from caffe operator and use pytorch's data interface.
Usage
First complile the source file
python setup.py build_ext --inplace
Then you can call the Rotated RoIAlign in python.
pooler_rotated=ROIAlignRotated((32,192), spatial_scale = (1.), sampling_ratio = 0)
image_roi_bbox=pooler_rotated(<Input image>,<Input RoI>)
Example
test.py
is a example how to use the rotated RoIAlign. The input image