Rotated_ROIAlign

Rotated RoIAlign in pytorch, implemented with CUDA. This implementation is edited from caffe operator and use pytorch's data interface.

Usage

1. First complile the source file

    python setup.py build_ext --inplace

2. 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 the roi result