• 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
onnxruntime_op.md 3.85 KB