add corner pooling (#2796)
* add center pooling
* fix yapf formatting
* add unittest
* fix flake8 formatting
* fix isort formatting
* fix isort formatting
* refactor and formatting
* remove top pool from unittext
* remove bottom pool from unittest
* fix bug in torch1.5
* add pytest for assertion
* center_pool -> corner_pool, add docstring
* fix unittest
* add corner_pool.py
* fix some weird logics
* move unittests of ops to a new folder
Co-authored-by:
Kai Chen <chenkaidev@gmail.com>
Showing
- mmdet/ops/__init__.py 35 additions, 9 deletionsmmdet/ops/__init__.py
- mmdet/ops/corner_pool/__init__.py 3 additions, 0 deletionsmmdet/ops/corner_pool/__init__.py
- mmdet/ops/corner_pool/corner_pool.py 101 additions, 0 deletionsmmdet/ops/corner_pool/corner_pool.py
- mmdet/ops/corner_pool/src/corner_pool.cpp 268 additions, 0 deletionsmmdet/ops/corner_pool/src/corner_pool.cpp
- setup.py 5 additions, 1 deletionsetup.py
- tests/test_ops/test_corner_pool.py 58 additions, 0 deletionstests/test_ops/test_corner_pool.py
- tests/test_ops/test_merge_cells.py 0 additions, 0 deletionstests/test_ops/test_merge_cells.py
- tests/test_ops/test_nms.py 0 additions, 0 deletionstests/test_ops/test_nms.py
- tests/test_ops/test_soft_nms.py 0 additions, 0 deletionstests/test_ops/test_soft_nms.py
- tests/test_ops/test_wrappers.py 0 additions, 0 deletionstests/test_ops/test_wrappers.py
Loading
Please register or sign in to comment