Support Better-Aligned ROIAlign Layer (#2068)
* feature (roi_align_v2): use roi_align v2 to support aligned=True
* fix (setup.py): fix install bug
* recover configs to the original ones
* remove cpu version for now
* rename files
* reformat (roi_align): clean code and reformat
* support legacy roi_align
* support legacy roi_align
* reformat (roi_align): reformat c/cuda/python code
* fix reformat request
* add unsaved change
* pass yapf
* merge roi_align function
* update note
Co-authored-by:
beansi <zhangwenwei@sensetime.com>
Showing
- mmdet/ops/roi_align/roi_align.py 84 additions, 19 deletionsmmdet/ops/roi_align/roi_align.py
- mmdet/ops/roi_align/src/roi_align_cuda.cpp 58 additions, 10 deletionsmmdet/ops/roi_align/src/roi_align_cuda.cpp
- mmdet/ops/roi_align/src/roi_align_kernel.cu 13 additions, 13 deletionsmmdet/ops/roi_align/src/roi_align_kernel.cu
- mmdet/ops/roi_align/src/roi_align_kernel_v2.cu 348 additions, 0 deletionsmmdet/ops/roi_align/src/roi_align_kernel_v2.cu
- setup.py 5 additions, 1 deletionsetup.py
Loading
Please register or sign in to comment