5 Commits (03491ed28a6f07c766b7a6950452b1782d95dae4)

Author	SHA1	Message	Date
q.yao	9ba1f76005	[Feature] : Add Deformable Conv2d TensorRT Plugin (#858) ... * add dcn tensorrt plugin * prepare for fp16 support * fix for lint * limit column buffer * add docstring to memcpyPermute	2021-03-11 15:41:47 +08:00
RunningLeon	3dd98daaa2	fix test unit of nms and batched_nms for tensorrt (#872)	2021-03-08 16:57:19 +08:00
q.yao	1ee5315ea2	[Feature] : Add NonMaxSuppression TensorRT Plugin (#787) ... * start trt plugin prototype * Add test module, modify roialign convertor * finish roi_align trt plugin * fix conflict of RoiAlign and MMCVRoiAlign * fix for lint * fix test tensorrt module * test_tensorrt move import to test func * add except error type * add tensorrt to setup.cfg * code format with yapf * fix for clang-format * move tensorrt_utils to mmcv/tensorrt, add comments, better test module * fix line endings, docformatter * isort init, remove trailing whitespace * add except type * fix setup.py * put import extension inside trt setup * change c++ guard, update pytest script, better setup, etc * sort import with isort * sort import with isort * move init of plugin lib to init_plugins.py * add scatternd, nms plugin (WIP) * fix bugs of trt_nms * add trt nms test module * fix bugs of scatternd * code optimize, add comment about nms kernel * fix transform_if bug of trt_nms_kernel * fix struct name * default nms offset=0, fix bugs of batched input * format with clang-format * onnx preprocess * much better nms implementation, no need to transfer memory between host and device * update preprocess_onnx * parse constant tensor from initializer in preprocess_onnx * update nms * remove unnecessary codes * workspace aligned address * format trt_plugin_helper.hpp * fix index memory bugs * set alignment to 16 by default * fix lint * fix nms offset * fix bugs of preprocess onnx * update test for nms * tensorrt only accept int32, not int64 * update nms comments * fix indexing for scores in nms * update trt temp * make trt-nms compatiable to #803 * fix lint * add docstring to trt_nms_kernel.cuda, add description to preprocess_onnx * add comment to score indexing * fix bugs of max output boxes Co-authored-by: maningsheng <maningsheng@sensetime.com>	2021-02-23 15:09:49 +08:00
q.yao	2ab544fc29	[Feature] : Add ScatterND TensorRT Plugin (#786) ... * add scatter plugin * fix bugs of scatternd * add trt scatternd plugin * format code with clang-format * add test for scatternd * skip test_tensorrt in CI * remove unused variable Co-authored-by: maningsheng <maningsheng@sensetime.com>	2021-01-20 11:15:07 +08:00
q.yao	0de9e149c0	[Feature]: Add custom operators support for TensorRT in mmcv (#686) ... * start trt plugin prototype * Add test module, modify roialign convertor * finish roi_align trt plugin * fix conflict of RoiAlign and MMCVRoiAlign * fix for lint * fix test tensorrt module * test_tensorrt move import to test func * add except error type * add tensorrt to setup.cfg * code format with yapf * fix for clang-format * move tensorrt_utils to mmcv/tensorrt, add comments, better test module * fix line endings, docformatter * isort init, remove trailing whitespace * add except type * fix setup.py * put import extension inside trt setup * change c++ guard, update pytest script, better setup, etc * sort import with isort * sort import with isort * move init of plugin lib to init_plugins.py * resolve format and add test dependency: tensorrt * tensorrt should be installed from source not from pypi * update naming style and input check * resolve lint error Co-authored-by: maningsheng <maningsheng@sensetime.com>	2021-01-06 11:05:19 +08:00