6 Commits

Author	SHA1	Message	Date
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