8 Commits (5afca135f71009a110d22a9786ad84394d120475)

Author	SHA1	Message	Date
RunningLeon	477f0c0a39	[Fix]: Set installation of onnx simplify's related packages to optional (#779) ... * Fix onnx, onnxruntime, onnxoptimizer import * Revert "Fix onnx, onnxruntime, onnxoptimizer import" This reverts commit 820e3785a1. * remove simplify from init file * update test for onnx simplify * update onnx doc	2021-01-08 13:15:37 +08:00
q.yao	0de9e149c0	[Feature]: Add custom operators support for TensorRT in mmcv (#686) ... * 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>	2021-01-06 11:05:19 +08:00
RunningLeon	83732ec2e3	[Feature] add onnx simplify (#751) ... * [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	2020-12-31 19:16:48 +08:00
tangyanf	741e52dd5e	[Feature]: add custom op roialign for onnxruntime (#724) ... * 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>	2020-12-29 14:09:30 +08:00
q.yao	94810f2297	[Feature]: Add custom operators support for onnxruntime in mmcv (#612) ... * 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>	2020-12-23 11:03:55 +08:00
q.yao	c390e327fa	add onnx support to roi_align and roi_pool (#593) ... * add onnx support to roi_align and roi_pool * add test on roi_align and roi_pool * fix conflict with master	2020-11-14 19:56:35 +08:00
Kai Chen	03214fd446	Support pytorch 1.7 and update the CI (#631) ... * support pytorch 1.7 and update the CI * fix CI * fix onnxruntime version in macos	2020-10-31 17:58:15 +08:00
robin Han	23b2bdbf52	add unittest for onnx convert (#608) ... * 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	2020-10-26 11:33:35 +08:00