mmcv/tests/test_ops
ZhangShilong 54a7ebb4ec
[Feature]: support Multi-Scale-DeformAttention in deformable-detr (#878)
* add c++ ms_deform_atten

* fix cpp lint

* fix cpp lint

* clang format

* remove cmakefile

* google style

* clang-format precommit

* use clang-format-lint-action

* add transformer base class

* add merge

* add docstr

* add pyargs

* fix according to commments

* resiger module

* change to use basemodule

* add _ between build function

* split the name

* fix according to comments

* fix lint and fix unitest

* fix cpp lint

* fix bug of deformdetr_atten

* fix drop out

* fix residual

* use CUDA_1D_KERNEL_LOOP
2021-04-23 16:35:15 +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 [Bug] Fix DeformConv2d bias error and add tests (#940) 2021-04-12 20:31:34 +08:00
test_deform_roi_pool.py Add new parrots extension implementation for all ops (#794) 2021-02-26 19:05:25 +08:00
test_focal_loss.py
test_fused_bias_leakyrelu.py [Feature] Add cuda ops: UpFirDn2d and fused_bias_leakyrelu (#900) 2021-03-21 21:23:09 +08:00
test_info.py
test_masked_conv2d.py
test_merge_cells.py
test_modulated_deform_conv.py
test_ms_deformable_attn.py [Feature]: support Multi-Scale-DeformAttention in deformable-detr (#878) 2021-04-23 16:35:15 +08:00
test_nms.py Add new parrots extension implementation for all ops (#794) 2021-02-26 19:05:25 +08:00
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 Rotated ROI align op for pytorch (cpu&cuda), parrots (cpu&cuda) and onnxruntime (cpu) (#933) 2021-04-19 15:04:01 +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_align_rotated.py [Feature]: Add Rotated ROI align op for pytorch (cpu&cuda), parrots (cpu&cuda) and onnxruntime (cpu) (#933) 2021-04-19 15:04:01 +08:00
test_roi_pool.py
test_saconv.py
test_syncbn.py
test_tensorrt.py [Feature] : Add grid_sampler TensorRT Plugin (#865) 2021-03-31 00:28:08 +08:00
test_tin_shift.py
test_upfirdn2d.py [Feature] Add cuda ops: UpFirDn2d and fused_bias_leakyrelu (#900) 2021-03-21 21:23:09 +08:00