7 Commits

Author	SHA1	Message	Date
q.yao	004c00675f	[Feature]: add modulated deformable conv TensorRT support (#1078) ... * add modulated dcn, better dcn plugin * clangformat * update documentation	2021-06-16 21:38:07 +08:00
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
RunningLeon	4d42365ac2	[Feature]: add TensorRT InstanceNormalization plugin (#1034) ... * add instancenorm plugin * resolve comments * fix lint * fix typo	2021-05-25 20:03:57 +08:00
lizz	13888df2aa	Fix typos (#1041) ... * Fix typos Signed-off-by: lizz <lizz@sensetime.com> * Add deprecation warning Signed-off-by: lizz <lizz@sensetime.com>	2021-05-24 14:51:45 +08:00
q.yao	9d1436fb6c	[Feature] add cummax/cummin tensorrt plugin (#1031) ... * add cummax/cummin tensorrt plugin * fix isort * fix with clang-format * fix with clang-format again * add document	2021-05-24 13:55:21 +08:00
q.yao	e478e9ff7f	[Doc]: Add document of TensorRT and onnxruntime custom ops (#920) ... * update tensorrt plugin document * add onnxruntime custom ops document * format document * add release note to onnxruntime_op and tensorrt_plugin * update document * add deployment.rst * add grid_sampler onnxruntime document * fix lint * add allow_different_nesting tag * add allow_different_nesting tag Co-authored-by: maningsheng <maningsheng@sensetime.com>	2021-04-12 14:06:28 +08:00
RunningLeon	5afca135f7	[Doc]: add a short document for TensorRT plugins in mmcv (#797) ... * add trt doc * add notes * add sphinx_markdown_tables * resolve comments * resolve comments	2021-01-28 15:57:13 +08:00