ff95bf48ea
[Feature] Support more MMOCR models: DBNet++, TextSnake, ABINet, MRCNN ( #1534 )
...
* WIP ocr
* add mrcnn rewrite
* add any rewrite for abinet
* export abinet to onnx
* fix abinet onnx export
* support abinet to tensorrt static and modify mmocr.yml
* add textsnake and dbnetpp
* support mrcnn in ORT and TRT
* add a condition before update data_preprocessor scope
* update doc and mmocr.yml
* add ut
* markdown and simple config
* write build_pytorch_model in child class
* update any_default
* remove where in abi_language_decoder___get_length__default
* keep where
* fix UT
* fix UT
* fix UT
* update mmocr.yml and config description
* tensorrt-fp32 -> tensorrt
* update doc
2023-01-16 12:06:03 +08:00
6b9bfcfceb
[Fix]: fix numba numpy version not compatiable in circileci ( #1611 )
...
* fix numba numpy version not compatibale
* fix mmcls ut
* update
* update
* only do lint when deploy configs changed
2023-01-11 10:48:52 +08:00
deaefacea1
Add softmax to cls model ( #1573 )
...
* Add softmax to cls model
* fix cls ci
* multihead
* update classification_model.py
2022-12-30 15:43:12 +08:00
d6fdb3e860
[Refactor] Add backend manager for 1.x ( #1515 )
...
* backend manager 1.x
* update pplnn init
* rename file
* add to backend
* add check env and misc
* fix action
* fix ut
* fix comment
2022-12-28 11:38:01 +08:00
6288141bd5
[Feature] Support RTMDet-R deployment ( #1553 )
...
* support rtmdet-r
* add comments
* fix dep config
* add ut
* fix ut
* fix ut
* fix url
* fix en url
2022-12-26 09:52:44 +08:00
e0ed95ebc4
[Fix] fix unittest and suppress warning ( #1552 )
...
* fix unittest and some warning
* fix read string
* snake
2022-12-22 19:57:36 +08:00
0e656067a6
[Enhancement] Support tvm 1.x ( #1531 )
...
* support tvm 1.x
* fix tvm sdk export
2022-12-22 00:15:47 +08:00
6e91614171
CodeCamp #102 : support SOLO deployment with OpenVINO ( #1454 )
...
* feat: support solo mmdet3.x openvino 2.0 api
* feat: support solo mmdet3.x openvino
* fix: lint
* fix: add solo head test
* docs: add supported_modesl
* docs: add supported_models
* fix: fix unreasonable code
* fix: fix ci failed
* feat: add linspace func rewrite
* fix: fix unreasonable rewrite linspace__onnx
* fix: change func name from __onnx to __default
* feat: add solo test regression
2022-12-22 00:02:25 +08:00
84ee45bc35
torchjit mdcn 1.x ( #1536 )
2022-12-20 15:13:49 +08:00
67c1cd2475
Add YOLOv5 support for RV1126 device. ( #1321 )
...
* support pose simcc
* fix yolov5 create_input
* add yolov5 post process and update mmcls.yml
* add letter resize
* add yolov5 export info
* fix
* add pose face config
* pick 4dd4d4851bhttps://github.com/open-mmlab/mmyolo/pull/305
* add pre_compile option
* update doc and fix typo
* fix padding
* fix typo
* use throw_exception
2022-12-16 17:05:24 +08:00
3f261e6d50
[Refactor] Refactor rewriter context for MMRazor ( #1483 )
...
* wip
* update rewriter
* Support all codebase
* update docs
* fix ssd
* rename qualname
* support torch.fx.wrap
* import by torch version
Co-authored-by: pppppM <gjf_mail@126.com>
2022-12-13 19:03:56 +08:00
78901a2451
cherry-pick #1353 and #1509
...
cherry-pick #1353 and #1509 and resolve the build issues
2022-12-13 02:53:01 -08:00
2afbb9945e
[Enhancement] MMRotate 1.x support ( #1401 )
...
* wip
* update twostage detector support
* fix unit test
* sdk wip
* comment
* refactor export info
* fix
* support roi trans
* update rotate.yml
* clear model.py, support torch1.13
2022-12-12 19:27:03 +08:00
04c9793da0
cherry-pick: Decouple preprocess operation and transformation ( #1353 )
2022-12-08 14:49:12 +08:00
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
AllentDan
RunningLeon
q.yao
Yanyi Liu
DDGRCF
q.yao
lvhan028
Li Zhang
hanrui1sensetime
SsTtOoNnEe
Chen Xin
vansin
tpoisonooo
Qingren
AllentDan