1. 24 Feb, 2021 2 commits
  2. 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 t...
      1ee5315e
  3. 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
  4. 21 Feb, 2021 2 commits
  5. 20 Feb, 2021 3 commits
  6. 18 Feb, 2021 2 commits
  7. 10 Feb, 2021 1 commit
  8. 07 Feb, 2021 4 commits
  9. 05 Feb, 2021 2 commits
  10. 04 Feb, 2021 2 commits
  11. 02 Feb, 2021 2 commits
  12. 01 Feb, 2021 1 commit
  13. 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
  14. 27 Jan, 2021 1 commit
  15. 21 Jan, 2021 1 commit
  16. 20 Jan, 2021 2 commits
  17. 14 Jan, 2021 1 commit
  18. 09 Jan, 2021 1 commit
  19. 08 Jan, 2021 5 commits
  20. 07 Jan, 2021 1 commit
  21. 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
  22. 04 Jan, 2021 1 commit
  23. 03 Jan, 2021 1 commit