1. 10 May, 2021 1 commit
    • v-qjqs's avatar
      Support custom operators cummax and cummin for onnxruntime (#1010) · 934b549e
      v-qjqs authored
      * support custom op `mmcv::cummax` for onnxruntime in mmcv
      
      * fix clang-format lint error
      
      * support mmcv::cummin, reformat codes
      
      * fix merge from master
      
      * add docs for mmcv::cummax and mmcv::cummin
      
      * format doc
      
      * add assertion for torch version, when exporting `cummax` to onnx
      
      * add more comments for torch version
      
      * handle exporting to onnx in `soft_nms`
      
      * commit for test_onnx
      
      * remove `is_in_onnx_export` in softnms
      
      * add more comments
      
      * fix c++ lint error
      
      * add known issues doc for `cummax`
      
      * fix known issues doc
      934b549e
  2. 01 May, 2021 1 commit
    • v-qjqs's avatar
      [Feature]: Support corner_pool related custom operators for onnxruntime in mmcv (#997) · db6b0542
      v-qjqs authored
      * supports for onnxruntime custom op `mmcv::MMCVTopPool`
      
      * supports for onnxruntime custom op `mmcv::MMCVCornerPool`, involving TopPool, BottomPool, LeftPool and RightPool
      
      * add unittest for corner_pool
      
      * supports mmcv::CornerPool without memcpy
      
      * add docs for mmcv::CornerPool
      
      * re-add docs for mmcv::CornerPool
      
      * fix output dtype doc
      
      * reformat
      
      * format with pre-commit
      
      * format
      
      * fix lint error, by using google clang-format style for c/c++
      db6b0542
  3. 12 Apr, 2021 1 commit
  4. 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
  5. 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