• Wang Xinjiang's avatar
    [Feature]: Add Rotated ROI align op for pytorch (cpu&cuda), parrots (cpu&cuda)... · ee041cec
    Wang Xinjiang authored
    
    [Feature]: Add Rotated ROI align op for pytorch (cpu&cuda), parrots (cpu&cuda) and onnxruntime (cpu) (#933)
    
    * add roi_align_rotated
    
    * code format
    
    * Add align key to roi align rotated
    
    * Add clockwise for rotated roi align
    
    * fix bugs in onnx export
    
    * Add docstring for RoIAlignRotated
    
    * remove cuda unittest
    
    * Reformat c++ code
    
    * add onnx roi align rotated file
    
    * fix unittest
    
    * Add  cpu and float64 of cuda support for parrots
    
    * code format
    
    * Add unified header to roi align rotated
    Co-authored-by: default avatarluopeichao <luopeichao@sensetime.com>
    ee041cec
test_roi_align_rotated.py 4.32 KB