1. 19 Apr, 2021 1 commit
    • 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