mmcv/tests/test_ops
Eugene Liu e621e08d54
Add DCN and Modulated DCN CPU implementation (#1278)
* DCN cpu version

* add modulated dcn cpu version

* move deform_conv_shape_check to deform conv utils

* add inline to deform_conv_shape_check

* add tests

* run linter

* add newline at file end

* run pre-commit against modulated deform conv cpp

* update saconv test

* run clang-format

* remove cuda device inline

* refactor dcn cuda/cpu functions

* remove DCN util

* remove DCN util hpp from all included files

* Addressing PR comment by refactoring modulated-DCN

* fix lint in cpp files
2021-08-29 20:48:31 +08:00
..
test_bbox.py
test_bilinear_grid_sample.py [Feature] enable exporting to onnx for PointRend (#953) 2021-06-11 13:49:19 +08:00
test_border_align.py Supports cuda version BorderAlign module (#1021) 2021-05-25 15:23:11 +08:00
test_box_iou_rotated.py
test_carafe.py
test_cc_attention.py
test_contour_expand.py Add pixel group and contour expand ops (#993) 2021-05-12 10:41:22 +08:00
test_corner_pool.py
test_deform_conv.py Add DCN and Modulated DCN CPU implementation (#1278) 2021-08-29 20:48:31 +08:00
test_deform_roi_pool.py
test_focal_loss.py
test_fused_bias_leakyrelu.py [Feature]: parrots add parrots/fused_bias & upfirdn2d (#989) 2021-04-27 13:49:50 +08:00
test_info.py
test_masked_conv2d.py
test_merge_cells.py
test_modulated_deform_conv.py Add DCN and Modulated DCN CPU implementation (#1278) 2021-08-29 20:48:31 +08:00
test_ms_deformable_attn.py Refactor the baseclass related to transformer (#978) 2021-06-11 18:09:31 +08:00
test_nms.py [Feature] NMS update (#957) 2021-05-31 23:42:26 +08:00
test_nms_rotated.py
test_onnx.py add modulated_deform_conv in onnxruntime support (#1281) 2021-08-23 11:22:24 +08:00
test_pixel_group.py Add pixel group and contour expand ops (#993) 2021-05-12 10:41:22 +08:00
test_psa_mask.py
test_roi_align.py
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 Add DCN and Modulated DCN CPU implementation (#1278) 2021-08-29 20:48:31 +08:00
test_syncbn.py
test_tensorrt.py [Feature]: support tensorrt custom plugin `MMCVCornerPool` (#1179) 2021-07-09 21:07:38 +08:00
test_tensorrt_preprocess.py [Feature] NMS update (#957) 2021-05-31 23:42:26 +08:00
test_tin_shift.py
test_upfirdn2d.py [Feature]: parrots add parrots/fused_bias & upfirdn2d (#989) 2021-04-27 13:49:50 +08:00