mirrors/mmdeploy
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmdeploy.git synced 2025-01-14 08:09:43 +08:00
Code Issues Projects Releases Wiki Activity
133 Commits 16 Branches 23 Tags
Commit Graph

11 Commits

Author SHA1 Message Date
Semyon Bevzyuk
c5a87fb1bc
[Enhancement]: OpenVINO deployment support for some models (#117) 
* Fix include and lib paths for onnxruntime.

* Fixes for SSD export test

* Add onnx2openvino and OpenVINODetector. Test models: ssd, retinanet, fcos, fsaf.

* Add support for two-stage models: faster_rcnn, cascade_rcnn

* Add doc

* Add strip_doc_string for openvino.

* Fix openvino preprocess.

* Add OpenVINO to test_wrapper.py.

* Fix

* Add openvino_execute.

* Removed preprocessing.

* Fix onnxruntime cmake.

* Rewrote postprocessing and forward, added docstrings and fixes.

* Added device type change to OpenVINOWrapper.

* Update forward_of_single_roi_extractor_dynamic_openvino and fix doc.

* Update docs.

* Add OpenVINODetector and onn2openvino tests.

* Add input_info to onnx2openvino.

* Add TestOpenVINOExporter and test_single_roi_extractor.

* Moved get_input_shape_from_cfg to openvino_utils.py and added test.

* Added test_cascade_roi_head.

* Add backend.check_env() to tests.

* Add OpenVINO to get_rewrite_outputs and to some tests in test_mmdet_models.

* Moved test_single_roi_extractor to test_mmdet_models.

* Removed TestOpenVINOExporter.
 2021-10-21 16:07:35 +08:00
lvhan028
6fdf6b8616
apply cmake-format and refactor cmake scripts of building onnx and te… (#99) 
* apply cmake-format and refactor cmake scripts of building onnx and tensorrt ops

* add static target

* suppress 'CMAKE_CUDA_ARCHITECTURES' warning when cmake with version 18 or later is used

* fix typo
 2021-10-13 16:42:11 +08:00
q.yao
f56a30025a
[Enhance] Add spell hook (#116) 
* add spell hook

* fix code spell
 2021-10-09 11:34:14 +08:00
AllentDan
8fe8056080
[Feature] Add trt ort mdcn plugins (#43) 
* add trt mdcn plugin

* add onnxruntime mdcn op

* add mdcn trt ort plugins

* fix lint

* remove comment

* remove plugin condition lines

* apply new form

* use serialized_size
 2021-08-25 10:17:21 +08:00
grimoire
77080bd931 fix onnxruntime roi_align on onnxruntime1.8 2021-07-26 15:57:13 +08:00
q.yao
04a8046864
[Feature] add register for ort custom op (#21) 
* add register for ort custom op

* ort_mmcv_util->ort_util

* ortApi->kOrtApi
 2021-07-23 17:16:07 +08:00
AllentDan
b9e64f9e1c
Add trt instance norm plugin (#16) 
* add trt instance norm plugin

* last line empty

* fix clang format

* fix grid_sample clang format

* remove redundant

* fix lint

* refine codes

* fix clang format

* clang format

* clang format

* clang format
 2021-07-16 16:05:09 +08:00
q.yao
3fa94f4424
[Feature] better tensorrt cpp code (#11) 
* better tensorrt cpp code

* fix end of file
 2021-07-12 10:01:52 +08:00
grimoire
8fd4ef4d47 remove unused flag, fix lint 2021-06-29 10:23:52 +08:00
grimoire
6eb2e89016 update cmakelists 2021-06-21 10:34:48 +08:00
grimoire
93408c7cd8 add type annotation 
add onnxruntime plugin support, add more custom ops

update function_rewriter, enable use origin function

add setup.cfg

update setup.cfg
 2021-06-16 15:36:58 +08:00