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
  2. 14 Nov, 2020 1 commit
  3. 31 Oct, 2020 1 commit
  4. 26 Oct, 2020 1 commit
    • robin Han's avatar
      add unittest for onnx convert (#608) · 23b2bdbf
      robin Han authored
      * add unittest for onnx convert
      
      * build onnx and onnxruntime in CI
      
      * skip onnx op unit test while using CUDA
      
      * fix offset==0 case in NMS
      
      * remove tmp file used in test
      
      * delete tmp file before assert so that we can remove the tmp file anyway
      23b2bdbf