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