1. 25 Feb, 2021 1 commit
  2. 24 Feb, 2021 2 commits
  3. 23 Feb, 2021 1 commit
    • q.yao's avatar
      [Feature] : Add NonMaxSuppression TensorRT Plugin (#787) · 1ee5315e
      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: default avatarmaningsheng <maningsheng@sensetime.com>
      1ee5315e
  4. 22 Feb, 2021 1 commit
    • ftbabi's avatar
      Add GELU activation function (#843) · 8735815a
      ftbabi authored
      * Support GELU activation function
      
      * Support different torch version
      
      * Fix bug in importing TORCH_VERSION
      
      * Wrap GELU module for PyTorch 1.3.x
      
      * Wrap GELU module for PyTorch 1.3.x
      8735815a
  5. 21 Feb, 2021 2 commits
  6. 20 Feb, 2021 3 commits
  7. 18 Feb, 2021 2 commits
  8. 10 Feb, 2021 1 commit
  9. 07 Feb, 2021 4 commits
  10. 05 Feb, 2021 2 commits
  11. 04 Feb, 2021 2 commits
  12. 02 Feb, 2021 2 commits
  13. 01 Feb, 2021 1 commit
  14. 28 Jan, 2021 2 commits
    • RunningLeon's avatar
      [Doc]: add a short document for TensorRT plugins in mmcv (#797) · 5afca135
      RunningLeon authored
      * add trt doc
      
      * add notes
      
      * add sphinx_markdown_tables
      
      * resolve comments
      
      * resolve comments
      5afca135
    • Haian Huang(深度眸)'s avatar
      Refactor _load_checkpoint fn (#790) · 96892bdc
      Haian Huang(深度眸) authored
      * Refactor _load_checkpoint fn
      
      * Update _load_checkpoint fn
      
      * Update docs str and add unit test
      
      * Fix unit test
      
      * Fix lint
      
      * Add comment and Optimize function
      
      * Fix docs str
      
      * Update load_ckpt and fix doc str
      
      * Update doc str and add sort unit test
      
      * Update and fix unit test
      
      * Fix unit test
      
      * Update and add unit test
      
      * Fix openmmlab prefix error
      96892bdc
  15. 27 Jan, 2021 1 commit
  16. 21 Jan, 2021 1 commit
  17. 20 Jan, 2021 2 commits
  18. 14 Jan, 2021 1 commit
  19. 09 Jan, 2021 1 commit
  20. 08 Jan, 2021 5 commits
  21. 07 Jan, 2021 1 commit
  22. 06 Jan, 2021 1 commit
    • q.yao's avatar
      [Feature]: Add custom operators support for TensorRT in mmcv (#686) · 0de9e149
      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: default avatarmaningsheng <maningsheng@sensetime.com>
      0de9e149
  23. 04 Jan, 2021 1 commit