SemyonBevzuk bf2c9fa8d2

[Feature] NMS update (#957) ...

* Add score_threshold and max_num to NMS

* Fix codestyle

* Fix codestyle

* Fix inds in nms

* Update nms docstring

* Move score_threshold and max_num arguments

* Fix args order in docstring

* fix lint of c++ file

* Remove torch.onnx.is_in_onnx_export() and add max_num to batched_nms for separate classes.

* Rewrote max_num handling in NMSop.symbolic

* Added processing max_output_boxes_per_class when exporting to TensorRT

* Added score_threshold and max_num for NMS in test_onnx.py and test_tensorrt.py

* Remove _is_value(max_num)

* fix ci errors with torch==1.3.1

* Update test_batched_nms in test_nms.py

* Added tests for preprocess_onnx

* Moved 'test_tensorrt_preprocess.py' and 'preprocess', updated 'remove_tmp_file'.

* Update mmcv/tensorrt/__init__.py

* Fix segfault torch==1.3.1 (remove onnx.checker.check_model)

* Returned 'onnx.checker.check_model' with torch version check

* Changed torch version from 1.3.1 to 1.4.0

* update version check

* remove check for onnx

Co-authored-by: maningsheng <maningsheng@sensetime.com>

2021-05-31 23:42:26 +08:00

..

test_bbox.py

Misc updates regarding Windows and unittests (#586)

2020-10-28 16:23: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

add 'iof' mode for box_iou_rotated (#753)

2020-12-25 17:06:45 +08:00

test_carafe.py

add ext ops, support parrots (#310)

2020-06-28 23:15:47 +08:00

test_cc_attention.py

add ext ops, support parrots (#310)

2020-06-28 23:15:47 +08:00

test_contour_expand.py

Add pixel group and contour expand ops (#993)

2021-05-12 10:41:22 +08:00

test_corner_pool.py

Migrate op (#392)

2020-07-08 17:29:15 +08:00

test_deform_conv.py

[Fix] Support amp (pytorch >= 1.6.0) on DCN and DCNv2/ Add unit tests on DCN/DCNv2 amp (#1029)

2021-05-23 17:42:59 +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

add ext ops, support parrots (#310)

2020-06-28 23:15:47 +08:00

test_fused_bias_leakyrelu.py

[Feature]: parrots add parrots/fused_bias & upfirdn2d (#989)

2021-04-27 13:49:50 +08:00

test_info.py

add ext ops, support parrots (#310)

2020-06-28 23:15:47 +08:00

test_masked_conv2d.py

add ext ops, support parrots (#310)

2020-06-28 23:15:47 +08:00

test_merge_cells.py

Migrate op (#392)

2020-07-08 17:29:15 +08:00

test_modulated_deform_conv.py

[Fix] Support amp (pytorch >= 1.6.0) on DCN and DCNv2/ Add unit tests on DCN/DCNv2 amp (#1029)

2021-05-23 17:42:59 +08:00

test_ms_deformable_attn.py

Add ms_deformable_attn in parrots (#1042)

2021-05-25 13:13:05 +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_nms.py

[Feature] NMS update (#957)

2021-05-31 23:42:26 +08:00

test_onnx.py

[Feature] NMS update (#957)

2021-05-31 23:42:26 +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

add ext ops, support parrots (#310)

2020-06-28 23:15:47 +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_align.py

fix roi_align ci for parrots (#708)

2020-12-13 20:01:25 +08:00

test_roi_pool.py

add ext ops, support parrots (#310)

2020-06-28 23:15:47 +08:00

test_saconv.py

fix saconv (#489)

2020-08-16 01:45:01 +08:00

test_syncbn.py

fix syncbn parameter order mismatch and parrots bug (#488)

2020-08-13 19:27:05 +08:00

test_tensorrt_preprocess.py

[Feature] NMS update (#957)

2021-05-31 23:42:26 +08:00

test_tensorrt.py

[Feature] NMS update (#957)

2021-05-31 23:42:26 +08:00

test_tin_shift.py

Add tin_shift function (#492)

2020-08-15 22:29:07 +08:00

test_upfirdn2d.py

[Feature]: parrots add parrots/fused_bias & upfirdn2d (#989)

2021-04-27 13:49:50 +08:00