7237b91dc5
[Fix]: Fix mmrotate and regression script ( #1917 )
...
* fix pipeline
* disable test for openvino and sdk+torchscript for mmpose
2023-03-28 12:11:35 +08:00
032ce75afa
mmclassification ConformerHead support ( #1905 )
...
* mmclassification ConformerHead support
* add mmclassification ConformerHead test config
---------
Co-authored-by: lishengxi <mtdp@MacBook-Pro-8.local>
2023-03-23 17:27:39 +08:00
d48187cf68
[Feature] Support RTMDet and RTMPose ncnn deployment ( #1857 )
...
* support rtmpose ncnn
* fix docformatter
* fix docformatter
* fix classname from tauj to dev-1.x branch
* rename file
* fix comments
* remove unused rewriter
* fix norm
* fix lint
* fix rtmcc_block
* fix norm
* add ut
* fix origin_func
* fix norm
* fix rtmdet_head
* add ut
* false run_with_backend for ncnn
* fix lint
2023-03-21 11:56:44 +08:00
50bd7e961e
support multi_scale_deform_attn trt plugin ( #1844 )
...
* support multi_scale_deform_attn trt plugin
* fix lint error
* add onnx symblic fun
* init msdeformablecrossattn symblic fun & fix plugin
* unittest for ms_deformable_cross_attn
* fix template
* update unittest
* fix input contiguous of trtwrapper
* update doc description
2023-03-17 15:51:02 +08:00
8e2f6556be
rewrite torch.cat for TensorRT when input is dynamic ( #1851 )
2023-03-17 15:49:52 +08:00
637958a910
[Fix] Fix MobileOne deployment. ( #1870 )
...
* fix task
* fix regression;
* fix ut
2023-03-13 10:54:07 +08:00
91a0a9af0b
[Fix] Fix unit test in dev1.x ( #1790 )
...
* fix erros in ut
* use nms_ratated ops instead of ext
* fix ut
* fix superresolution
* update super_resolution
2023-03-08 14:32:50 +08:00
bf36950f0e
[Fix] fix centernet ( #1768 )
...
* fix centernet
* update sdk transform
2023-02-24 10:50:03 +08:00
d0f8be63f3
fix ut and text_det create_input ( #1756 )
2023-02-20 16:45:20 +08:00
1f56eea807
[Feature] Support feature map output for mmsegmentation ( #1625 )
...
* add feature map output for mmseg
* update api
* update demo
* fix return
* update format_shape
* fix lint
* update csharp demo
* update python demo && api
* fix coreml build
* fix lint
* better sort
* update
* update cpp demo & add missing header
* change to CHW
* update csharp api
* update isort version to 5.12.0
* fix python api
* fix log
* more detail api docs
* isort support python3.7
* remove isort change
* remove whitespace
* axes check
* remove FormatShapeImpl
* minor
* add permute tc
* remove stride buffer
(cherry picked from commit b85f34141b
2023-02-07 21:04:20 +08:00
dc1d9df315
[Fix] fix coreml for branch 1.x ( #1669 )
...
* fix coreml for branch 1.x
* fix docstring
* update docsting1
2023-01-31 13:49:48 +08:00
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
RunningLeon
Shengxi Li
hanrui1sensetime
Shaun
AllentDan
q.yao
Chen Xin
Yanyi Liu
DDGRCF
q.yao
lvhan028
Li Zhang
SsTtOoNnEe
vansin
tpoisonooo