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.py	[Feature] NMS update (#957)	2021-05-31 23:42:26 +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] 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.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	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.py	[Feature] NMS update (#957)	2021-05-31 23:42:26 +08:00
test_tensorrt_preprocess.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