1. 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