070036f964
[Refactor] better SDK registry ( #1368 )
...
* refactor SDK registry
* fix lint
* fix typo
* sync
* use nested namespace
* rename
(cherry picked from commit 99040d5655
2022-12-05 11:44:44 +08:00
3cb4a3fc6a
init ( #1423 )
2022-11-25 11:55:24 +08:00
71344b27a8
update reg test ( #1417 )
...
* give model path if stead of 'x' when conversion failed
* set --models with default value ['all']
* fix mmseg yml
2022-11-24 14:04:51 +08:00
cb37b092bd
update symbolic rewriter 1.x ( #1404 )
...
* update symbolic rewriter 1.x
* typo fix
2022-11-23 10:18:25 +08:00
a10b9e964b
Fix some bugs on dev1.x ( #1390 )
...
* fix onnx export unused param
* add cfgoptions in reg test
2022-11-18 18:35:00 +08:00
115ca470dd
fix mobileone regtest ( #1393 )
2022-11-18 16:43:56 +08:00
81eafa5fb3
[Fix] fix regression test for mmocr( #1372 )
...
* fix
* refine
2022-11-18 11:37:26 +08:00
680a3fb239
Support MobileOne on ONNXRuntime and TensorRT ( #1268 )
...
* Support MobileOne
* Updated supported model readme (Add MobileOne)
* Update supported_models.md
* Updated supported model readme (Add MobileOne)
* Update mmcls.yml
* Support MobileOne
* Remove other files
* Resolve lint error
* Remove converted model files
* Remove json files
* Update mmcls.yml
* Update mmcls.yml
* Update benchmark doc
Co-authored-by: SenseTime Research Singapore <SENSETIME\research.sgres@sg0016000001u.domain.sensetime.com>
2022-11-14 17:46:42 +08:00
83756b97c6
[Feature] Support centernet dev1.x ( #1219 )
...
* support centernet head
* add centernet head ut
* add centernet
* add centernet
* add support models
* fix mdformat
* fix reg test
* fix scale
* fix test.py show_dir kwargs
* fix for profile in T4
* fix dynamic shape
* fix lint
* move rescale and border to outside
* fix ut
* fix lint
* update ort torchscript benchmark
* fix centernet
* fix ut
* remove unused file
* support centernet sdk
* remove unused rewriter
* fix lint
* fix flake8
* remove unused line
* fix lint
* fix lint
* fix doc links
* fix mdformat
* fix scale_factor as default
* apart random pad and pad
* fix sdk
* fix centernet docs
* fix code style of cpp
2022-11-14 16:53:05 +08:00
8b5a6dd7eb
[Fix] fix test ops ( #1352 )
...
* fix test ops
* fix name
2022-11-14 11:19:57 +08:00
ccc21289d1
[Feature] Sync mmaction2-sdk(master) to dev1.x ( #1307 )
...
* sync mmaction sdk
* pipeline.json
* fix docs
* replace topk when make regression for mmaction2
* add python api
* add missing file
* add missing test file
* remove cudnn dep for formatshape
* add sample arg for input
2022-11-10 15:13:24 +08:00
ab88248cca
[Fix] Fix config name in mmdet yaml ( #1334 )
...
* [Fix] fix the yolox yolov3 config name
* [Fix] set ncnn backend_test False
2022-11-09 14:11:42 +08:00
0dc25a27dd
fix(circleci): rewrite outputs is a list ( #1319 )
2022-11-08 17:26:36 +08:00
d330e17af3
fix reg of dev-1.x( #1317 )
2022-11-08 11:17:59 +08:00
6420e20445
[Refactor] Ease rewriter import for dev-1.x ( #1170 )
...
* Update rewriter import
* remove root import
* add interface
2022-11-08 10:37:59 +08:00
c5edb85550
Sync rv1126 to dev-1.x by cherry-pick ( #1295 )
...
* remove imports (#1207 )
* remove imports
* update doc
* detailed docstring
* rephrase
* Add model conversion support to rv1126 (#1203 )
* WIP
* fix interpolate
* support yolov3 and retinanet
* support seg
* support ssd
* supports both partition types for retinanet and ssd
* mean std doc
* update doc, add UT
* support FSAF
* rename configs
* update dump info
* update
* python package installation doc
* update doc
* update doc
* doc
* fix
* docstring
* remote partition config
2022-11-07 10:19:22 +08:00
331292a992
Feature: support mmdet3d dev-1.x 1.1.0rc1 ( #1225 )
...
* feat(mmdet3d): test pointpillars and centerpoint on ort, openvino and trt passed
* fix(centerpoint): mvx_two_stage input error
* fix(review): remove mode decorator
* fix(mmdet3d): review advices
* fix(regression): update mmdet3d.yml and test ort/openvino passed
* unittest(mmdet3d): fix
* fix(unittest): fix
* fix(mmdet3d): unittest
* fix(mmdet3d): unittest
* fix(CI): remove mmcv.Config
* fix(mmdet3d): unittest
* fix(mmdet3d): support torch1.12
* fix(CI): use bigger point cloud file
* improvement(mmdet3d): align backend outputs with torch
* fix(mmdet3d): remove useless
* style(mmdet3d): format code
* style(mmdet3d): remove useless
* fix(mmdet3d): sync vis_task
* unittest(mmdet3d): add test
* docs(mmdet3d): add docstring
* unittest(ci): add unittest data
* fix(mmdet3d): review advices
* feat(mmdet3d): convert fail
* style(mmdet3d): docstring
* style(mmdet3d): docstring
2022-11-04 20:54:01 +08:00
13290614f6
Fix hourglass from mmpose ( #1277 )
...
* update mmpose rewritings
* update yml
* update docstring for mmpose
2022-11-03 15:15:33 +08:00
1d47848184
fix ci error
2022-11-01 18:54:29 +08:00
2fa5095154
cherry-pick from commit 197a7ad
2022-11-01 16:40:12 +08:00
53d4668a2f
[Feature] Add `Cond` node and pose tracker demo ( #1186 )
...
* add Cond node
* WIP PoseTracker
* fix pose tracker
* minor fix
* simplify design
* add timing
* sync
* visualize
* remove file check
(cherry picked from commit f2be2abeb5
2022-11-01 16:36:43 +08:00
4b39205866
Rewrite Conv2dAdaptiveOps for conversion of EfficientNet (static shape) ( #1045 )
...
* Rewrite Conv2dAdaptiveOps for conversion of EfficientNet (static shape)
* Refactor codes and Add unit test
* Simplify codes
* update supported model configs in yaml
* update mmcls.yml
* resolve lint error
Co-authored-by: SenseTime Research Singapore <SENSETIME\research.sgres@sg0016000001u.domain.sensetime.com>
(cherry picked from commit dd7550a08d
2022-11-01 16:29:15 +08:00
0efc9e3c6d
[Feat]: Support simcc from mmpose ( #1187 )
...
* add rewriting for simcc
* add simccdecode for sdk
* remove debug lines
* fix cpp lint
* move simcc decode to sdk
* add simcc sdk config
* update docs and regression yaml
* update yml
2022-10-27 11:42:50 +08:00
2020e74480
Support mmaction2:dev-1.x ( #1012 )
...
* support tsn
* support slowfast
* fix export info & End2EndModel
* add test
* fix forward
* fix lint
* update tests
* add onnxruntime 2d config
* fix ort-gpu
* add mmaction.yml, need to update
* fix reviews
* add ann.txt
* add visualize
* fix lint
* rebase
* add conftest.py
* fix circle ci
* fix registry
* fix regression test
2022-10-19 15:42:57 +08:00
72923e7844
update mmpose ( #1213 )
2022-10-18 10:24:19 +08:00
6e9c2527b4
merge master from c35099ef0
2022-10-13 11:12:07 +08:00
fd21b98efa
[Docs] add the support information about MMPose Hourglass ( #1118 )
...
* [Docs] add the support information about MMPose Hourglass
* * add Hourglass support config
* update benchmark doc
* modify Hourglass AP under pytorch backend
* add regression test for Hourglass
* *update 03-benchmark/benchmark.md
* * modify mmpose.yml to ignore ncnn int8 test
* add trt_fp16 test results
* * add openvino test result
* * modify supported_models.md
2022-10-10 16:38:54 +08:00
41fcc2242f
log cmd info for regresstion test ( #1146 )
...
* add log to regresstion test
* update for win
* update mmseg reg yml
2022-10-10 16:23:24 +08:00
51f4e65185
fix reg test for 2.0 ( #1116 )
...
* fix reg
fix mmedit mmocr
update
update apcnet of mmseg
catch error when download fails
log error message
* update reg for win
* update yml
2022-10-08 19:14:33 +08:00
2e459dba6d
support mmpose2.0 for sdk ( #1080 )
...
* add mmpose sdk
* --amend
* remove debug draw code
* fix docstring
* update
2022-10-08 15:49:44 +08:00
b78121a776
fix conflicts with dev-1.x
2022-09-30 10:31:32 +08:00
161f27d493
fix mmedit test picture ( #1131 )
2022-09-29 17:13:23 +08:00
23e84f46ff
update mmcls yaml ( #1137 )
2022-09-29 17:09:17 +08:00
7f70d7fe56
[Enhancement 2.0] mmdeploy for mmyolo ( #1088 )
...
* support for external codebase like mmyolo
* support for external export
* fix missing flake8
* fix comments
* add aenum
* add missing files
* fix condition
* refactor import_codebase
* fix mmyolo support
* fix lint
* add base codebase
* fix a strange clang-format
* fix import_codebase
* fix dependent codebase register
* wrap custom_model
* fix comment
* add ut
2022-09-29 16:40:47 +08:00
4fc8828af8
fix ocr UT
...
fix ci and lint
fix det
fix cuda ci
fix mmdet test
update object detection
fix ut
fix layer norm ut
update ut
lock mmeit version
fix mmocr mmcls ut
add conftest.py
fix ocr ut
fix mmedit ci
install mmedit from source
fix rknn model and prepare_onnx_paddings__tensorrt UT
docstring
fix coreml export
update mmocr config
small test
recovery assert
fix ci
2022-09-29 16:37:36 +08:00
144fcf7b3a
fix github ci
...
add mim
update circle ci
fix tests
fix ci
2022-09-29 16:26:31 +08:00
d282503012
update detyaml ( #1133 )
2022-09-29 12:14:06 +08:00
d7cbb395da
[Enhancement 2.0] mmdeploy for mmyolo ( #1088 )
...
* support for external codebase like mmyolo
* support for external export
* fix missing flake8
* fix comments
* add aenum
* add missing files
* fix condition
* refactor import_codebase
* fix mmyolo support
* fix lint
* add base codebase
* fix a strange clang-format
* fix import_codebase
* fix dependent codebase register
* wrap custom_model
* fix comment
* add ut
2022-09-28 16:30:29 +08:00
bf887cc8bc
[Enhancement] Refactor SDK pipeline ( #938 )
...
* unify C API naming
* fix demo and move apis/c/* -> apis/c/mmdeploy/*
* fix lint
* WIP refactor pipeline
* backward compatibility
* working pipeline demo
* add text det-recog demo
* add det-pose demo
* fix build
* fix demo
* add environment interface
* add environment to pass scheduler & model info at runtime
* update demos
* add pipeline API for Python
* fix `FromPyObject`
* fix for opencv-4.2
* environment -> context, improve pipeline
* python model interface
* fix cmake
* fix python & cmake
* context & C++ pipeline API
* minor fix
* improve API
* fix shared libs
* refresh C/Python API
* propagate context
* fix python demo
* fix
* add namespace
* fix namespace
* fix mis-changed strings
* fix
* fix python api
* rename
* clean-up
* fix pose detector
* clean-up
* clean-up
* clean-up
* fix python API build
* fix CI
* fix lint
* fix lint
* fix lint & demo
* install pipeline.hpp
* fix MSVC shared library build
* fix sample
* fix MSVC monolithic build
* minor fix
2022-09-26 16:11:14 +08:00
40adb329a3
Merge master:ea7706cb into sync_master
2022-09-22 19:49:50 +08:00
ea7706cbfd
fix layer norm ( #1015 )
2022-09-21 16:24:47 +08:00
e3859004ba
add mod rewriter for TRT ( #1024 )
2022-09-21 14:47:36 +08:00
5c87dd9565
[2.0] Support mmedit 2.0 ( #1017 )
...
* mmcv.Config -> mmengine Config
* support mmedit part
* add rewriter for BaseEditModels
* fix visualizer
* mmedit visualization
* remove unused code
* fix realesrgan
* fix trt
* support MultiTestLoop; rewriter fix mmediting bugs; fix ut
* fix uts
* fix mmedit sdk
* fix regression test(part)
* fix torchscript
* part of fix regression test
* fix checkenv.py
* fix test.py for mmedit2.0
* support for mmedit
* fix regression_test
* fix check copyright ci
* fix isort
* fix docformatter
* fix yapf
* fix tests
* fix sdk after 1040
* add a file for ut
* fix docformatter
* fix export info
* fix super_resolution
* fix test.py
* stage configs
* remove unused code
* remove rewriter of multitestloop
* fix yapf
2022-09-20 19:22:55 +08:00
2fdba2c523
Support mmpose:dev-1.x ( #884 )
...
* modify mmpose rewritings
* test exporting to ort with hrnet
* support mspn
* update tests
* update tests
* support flip_test
* support mmpose 1.x
* update mmpose.yml
* fix adaptive pool
* align with master for adaptive_pool rewriting
* fix pipeline
* add batch_size to test
* resolve comment
2022-09-19 21:40:53 +08:00
97e0d1228f
SDK ocr 2.0 ( #1006 )
...
* add deploy runner
* fix text_det wrapper
* fix recog
* save
* add crnn support
* update with_padding
* add short scale aspect jitter
* update regression test
* torch2ts
* add test data
* resolve comments
2022-09-19 15:08:51 +08:00
0caeaf238c
[CustomOps] TensorRT Gather Topk Ops ( #1033 )
...
* add gather topk
* add shape inference and document
* fix faster rcnn
* reshape topk
* fix
2022-09-19 13:48:26 +08:00
b0b502cba0
[Fix] Sync codebase 2.0 ( #979 )
...
* remove mmcv.Config
* fix mmdet with new mmengine
* fix ut
* fix mmdet ut
* fix sdk cpp pipeline
* fix mmcls
* fix mmcls ut
* adapt for new config
* fix cls models
* fix mmcls sdk
* fix yapf
* fix lint
* fix torchscript
* fix torchscript cpu
* fix check_env
* add test data
* fix export_info
* fix blank.jpg
* fix mmdet torchscript
* fix test.py
* fix sdk forward
* fix yapf
2022-09-15 21:54:57 +08:00
06028d6a21
Support mmseg:dev-1.x ( #790 )
...
* support pspnet + ort
* add rewriting for adapt_avg_pool
* test pspnet
* resize seg_pred to original image shape
* run with test.py
* keep as original
* fix ut of segmentation
* update var name
* fix export to torchscript
* sync with mmseg:test-1.x branch
* fix ut
* fix regression test for mmseg
* fix mmseg.ops
* update mmseg yml
* fix mmseg2.0 sdk
* fix adaptive pool
* update rewriting and tests
* fix sdk inputs
2022-09-14 20:08:52 +08:00
124635ec5f
Add RKNN support. ( #865 )
...
* save codes
* support resnet and yolov3
* support yolox
* fix lint
* add mmseg support and a doc
* add UT
* update supported model list
* fix ci
* refine docstring
* resolve comments
* remote output_tensor_type
* resolve comments
* update readme
2022-09-06 11:48:39 +08:00
6b01a2e649
[Feature] Add option to fuse transform. ( #741 )
...
* add collect_impl.cpp to cuda device
* add dummy compute node wich device elena
* add compiler & dynamic library loader
* add code to compile with gen code(elena)
* move folder
* fix lint
* add tracer module
* add license
* update type id
* add fuse kernel registry
* remove compilier & dynamic_library
* update fuse kernel interface
* Add elena-mmdeploy project in 3rd-party
* Fix README.md
* fix cmake file
* Support cuda device and clang format all file
* Add cudaStreamSynchronize for cudafree
* fix cudaStreamSynchronize
* rename to __tracer__
* remove unused code
* update kernel
* update extract elena script
* update gitignore
* fix ci
* Change the crop_size to crop_h and crop_w in arglist
* update Tracer
* remove cond
* avoid allocate memory
* add build.sh for elena
* remove code
* update test
* Support bilinear resize with float input
* Rename elena-mmdeploy to delete
* Introduce public submodule
* use get_ref
* update elena
* update tools
* update tools
* update fuse transform docs
* add fuse transform doc link to get_started
* fix shape in crop
* remove fuse_transform_ == true check
* remove fuse_transform_ member
* remove elena_int.h
* doesn't dump transform_static.json
* update tracer
* update CVFusion to remove compile warning
* remove mmcv version > 1.5.1 dep
* fix tests
* update docs
* add elena use option
* remove submodule of CVFusion
* update doc
* use auto
* use throw_exception(eEntryNotFound);
* update
Co-authored-by: cx <cx@ubuntu20.04>
Co-authored-by: miraclezqc <969226879@qq.com>
2022-09-05 20:29:18 +08:00
Li Zhang
AllentDan
RunningLeon
q.yao
hanrui1sensetime
SsTtOoNnEe
Chen Xin
vansin
tpoisonooo
lvhan028
Qingren
AllentDan
Chen Xin