1. 18 Feb, 2021 2 commits
  2. 10 Feb, 2021 1 commit
  3. 07 Feb, 2021 4 commits
  4. 05 Feb, 2021 2 commits
  5. 04 Feb, 2021 2 commits
  6. 02 Feb, 2021 2 commits
  7. 01 Feb, 2021 1 commit
  8. 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
  9. 27 Jan, 2021 1 commit
  10. 21 Jan, 2021 1 commit
  11. 20 Jan, 2021 2 commits
  12. 14 Jan, 2021 1 commit
  13. 09 Jan, 2021 1 commit
  14. 08 Jan, 2021 5 commits
  15. 07 Jan, 2021 1 commit
  16. 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
  17. 04 Jan, 2021 1 commit
  18. 03 Jan, 2021 1 commit
  19. 02 Jan, 2021 1 commit
  20. 31 Dec, 2020 1 commit
    • RunningLeon's avatar
      [Feature] add onnx simplify (#751) · 83732ec2
      RunningLeon authored
      * [Enhance]: add onxx simplify
      
      * add simple doc
      
      * add unit test
      
      * update docstring
      
      * resolve some comment
      
      * add test dependency:onnxoptimizer
      
      * Fix onnxruntime register empty libpath
      
      * test onnxruntime version
      
      * set checker to false
      
      * skip test_simplify for torch<1.5.0
      83732ec2
  21. 29 Dec, 2020 1 commit
    • tangyanf's avatar
      [Feature]: add custom op roialign for onnxruntime (#724) · 741e52dd
      tangyanf authored
      
      
      * add ort custom op roialign
      
      * Update roi_align.py
      
      * Update test_onnx.py
      
      * Update test_onnx.py
      
      * Update test_onnx.py
      
      * Update test_onnx.py
      
      * Update onnxruntime_register.cpp
      
      * Update roiAlign.h
      
      * Update roiAlign.cpp
      
      * lint modification
      
      * update roiAlign.cpp
      
      * lint check
      
      * lint check
      
      * lint fix
      
      * lint fix
      
      * fix lint
      
      * add link to commit
      Co-authored-by: default avatarmaningsheng <maningsheng@sensetime.com>
      741e52dd
  22. 28 Dec, 2020 3 commits
  23. 26 Dec, 2020 1 commit
  24. 25 Dec, 2020 2 commits