1. 01 Feb, 2021 1 commit
  2. 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
  3. 27 Jan, 2021 1 commit
  4. 21 Jan, 2021 1 commit
  5. 20 Jan, 2021 2 commits
  6. 14 Jan, 2021 1 commit
  7. 09 Jan, 2021 1 commit
  8. 08 Jan, 2021 5 commits
  9. 07 Jan, 2021 1 commit
  10. 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
  11. 04 Jan, 2021 1 commit
  12. 03 Jan, 2021 1 commit
  13. 02 Jan, 2021 1 commit
  14. 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
  15. 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
  16. 28 Dec, 2020 3 commits
  17. 26 Dec, 2020 1 commit
  18. 25 Dec, 2020 2 commits
  19. 24 Dec, 2020 4 commits
  20. 23 Dec, 2020 1 commit
    • q.yao's avatar
      [Feature]: Add custom operators support for onnxruntime in mmcv (#612) · 94810f22
      q.yao authored
      
      
      * add onnx support to roi_align and roi_pool
      
      * add softnms ort support
      
      * fix for lint
      
      * format cpp code with clang-format:google
      
      * add new empty line to the end of head files in onnxruntime
      
      * update to pytorch1.7
      
      * add test of softnms to onnxruntime
      
      * fix for lint
      
      * remote print in ops/info.py
      
      * change import order, fix for flake8
      
      * fix include
      
      * add assert torch>=1.7.0
      
      * [doc]: add document for onnxruntime custom operator
      
      * update onnxruntime version to v1.5.1 for softnms
      
      * remove doc menu
      
      * Resolve lint for markdown
      
      * resolve naming style in onnxruntime_op.md
      
      * Use old cpp apis, optimize test_onnx.py
      
      * Fixing strings in tests/test_ops/test_onnx.py
      
      * code format with yapf
      
      * fix soft_nms parrot
      
      * add import in onnxruntime setup, avoid conflict
      
      * fix doc and add assert
      
      * change cpp guard
      Co-authored-by: default avatarmaningsheng <maningsheng@sensetime.com>
      94810f22
  21. 22 Dec, 2020 4 commits
  22. 21 Dec, 2020 2 commits
  23. 20 Dec, 2020 1 commit
    • Wang Xinjiang's avatar
      Allow to replace nested tuple and list via options (#731) · 96ebfa65
      Wang Xinjiang authored
      * Allow to replace nested tuple and list via options
      
      * Add comments
      
      * Fix single nested items
      
      * Simplify the code
      
      * Simplify the code
      
      * Simplify the code
      
      * Simplify the code
      
      * Update docstring
      
      * Update docstring
      
      * Support quotation mark
      
      * modify docstring
      96ebfa65
  24. 18 Dec, 2020 1 commit