looput / Rotated_ROIAlign

Rotated ROIAlign in pytorch, implemented with CUDA

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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
  1. 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 input image the roi result output image

About

Rotated ROIAlign in pytorch, implemented with CUDA

License:MIT License


Languages

Language:Cuda 66.9%Language:Python 26.1%Language:C++ 7.0%