mmcv/tests/test_ops
q.yao 0de9e149c0
[Feature]: Add custom operators support for TensorRT in mmcv (#686)
* start trt plugin prototype

* Add test module, modify roialign convertor

* finish roi_align trt plugin

* fix conflict of RoiAlign and MMCVRoiAlign

* fix for lint

* fix test tensorrt module

* test_tensorrt move import to test func

* add except error type

* add tensorrt to setup.cfg

* code format with yapf

* fix for clang-format

* move tensorrt_utils to mmcv/tensorrt, add comments, better test module

* fix line endings, docformatter

* isort init, remove trailing whitespace

* add except type

* fix setup.py

* put import extension inside trt setup

* change c++ guard, update pytest script, better setup, etc

* sort import with isort

* sort import with isort

* move init of plugin lib to init_plugins.py

* resolve format and add test dependency: tensorrt

* tensorrt should be installed from source not from pypi

* update naming style and input check

* resolve lint error

Co-authored-by: maningsheng <maningsheng@sensetime.com>
2021-01-06 11:05:19 +08:00
..
test_bbox.py
test_box_iou_rotated.py add 'iof' mode for box_iou_rotated (#753) 2020-12-25 17:06:45 +08:00
test_carafe.py
test_cc_attention.py
test_corner_pool.py
test_deform_conv.py
test_deform_roi_pool.py
test_focal_loss.py
test_info.py
test_masked_conv2d.py
test_merge_cells.py
test_modulated_deform_conv.py
test_nms.py
test_nms_rotated.py Fix some bug as well as unit test for nms_rotate (#728) 2020-12-18 15:17:11 +08:00
test_onnx.py [Feature]: Add custom operators support for TensorRT in mmcv (#686) 2021-01-06 11:05:19 +08:00
test_psa_mask.py
test_roi_align.py fix roi_align ci for parrots (#708) 2020-12-13 20:01:25 +08:00
test_roi_pool.py
test_saconv.py
test_syncbn.py
test_tensorrt.py [Feature]: Add custom operators support for TensorRT in mmcv (#686) 2021-01-06 11:05:19 +08:00
test_tin_shift.py