- 09 Jul, 2021 1 commit
-
-
qjqfl authored
-
- 16 Jun, 2021 1 commit
-
-
q.yao authored
* add modulated dcn, better dcn plugin * clangformat * update documentation
-
- 31 May, 2021 1 commit
-
-
SemyonBevzuk authored
* 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>
-
- 25 May, 2021 1 commit
-
-
RunningLeon authored
* add instancenorm plugin * resolve comments * fix lint * fix typo
-
- 24 May, 2021 1 commit
-
-
q.yao authored
* add cummax/cummin tensorrt plugin * fix isort * fix with clang-format * fix with clang-format again * add document
-
- 30 Mar, 2021 1 commit
-
-
q.yao authored
* add grid sample trt support * fix align_corners=True * change 'intep' to 'interp', fix bugs of grid sampler, better test script * remove unused import * move source comment to the top of trt_grid_sampler_kernel.cu
-
- 11 Mar, 2021 1 commit
-
-
q.yao authored
* add dcn tensorrt plugin * prepare for fp16 support * fix for lint * limit column buffer * add docstring to memcpyPermute
-
- 08 Mar, 2021 1 commit
-
-
RunningLeon authored
-
- 23 Feb, 2021 1 commit
-
-
q.yao authored
* 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:
-
- 20 Jan, 2021 1 commit
-
-
q.yao authored
* 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:
-
- 06 Jan, 2021 1 commit
-
-
q.yao authored
* 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:
-
