- 26 Feb, 2021 2 commits
-
-
z55250825 authored
* delete all parrots file add bbox_overlaps new parrots op impl * support first new impl parrts op (bbox_overlaps)(success test) * add box_iou_rotated op, test succeed * add carafe and carafe_naive op, test succeed (one parrots bug need fix) * add cc_attention op, test success * add corner_pool op, test success * add parrots op deform_conv, test success * add deform_roi_pool op, test success (but has question) * add focal loss op, test success (gradcheck) * add masked_conv2d op, test success * add modulated_deform_conv op, test success * add nms and nms_rotated op, test success * add psamask op, test success * add roi_align op, test_success * add roi_pool op, test success * add sync_bn op, test success * add tin_shift op, test success * fix test_deform_roi_pool, add parrots test * skip test_onnx because parrots does not support onnx * fix c++ lint * fix python lint * fix python lint
-
RunningLeon authored
* Fix pytorch2onnx for yolov3 with torch==1.6.0 * update and add test for F.interpolate * add test_onnx.py with build_cuda
-
- 04 Feb, 2021 1 commit
-
-
tangyanf authored
* add ort nms op * fit lint check * fix lint check * update code * fix lint check * update code * update code * update code * update code * update code * update code
-
- 08 Jan, 2021 1 commit
-
-
RunningLeon authored
* Fix onnx, onnxruntime, onnxoptimizer import * Revert "Fix onnx, onnxruntime, onnxoptimizer import" This reverts commit 820e3785a1144eb85e1ba22688dada681ea4c829. * remove simplify from init file * update test for onnx simplify * update onnx doc
-
- 06 Jan, 2021 1 commit
-
-
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:maningsheng <maningsheng@sensetime.com>
-
- 31 Dec, 2020 1 commit
-
-
RunningLeon authored
* [Enhance]: add onxx simplify * add simple doc * add unit test * update docstring * resolve some comment * add test dependency:onnxoptimizer * Fix onnxruntime register empty libpath * test onnxruntime version * set checker to false * skip test_simplify for torch<1.5.0
-
- 29 Dec, 2020 1 commit
-
-
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:maningsheng <maningsheng@sensetime.com>
-
- 23 Dec, 2020 1 commit
-
-
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:maningsheng <maningsheng@sensetime.com>
-
- 14 Nov, 2020 1 commit
-
-
q.yao authored
* add onnx support to roi_align and roi_pool * add test on roi_align and roi_pool * fix conflict with master
-
- 31 Oct, 2020 1 commit
-
-
Kai Chen authored
* support pytorch 1.7 and update the CI * fix CI * fix onnxruntime version in macos
-
- 26 Oct, 2020 1 commit
-
-
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
-