Commit Graph

65 Commits (6cd29e2152d6935bde2f9252b47170724bac20ac)

Author SHA1 Message Date
huayuan4396 cbfe99505c
Fit for MMPretrain EfficientFormer (#2108)
* add ef in rgtest metafile

* fix for EfficientFormerClsHead to LinearClsHead

* fix lint

* update benchmark

* fix lint

* fix docs

* fix docs

* add openvino test
2023-06-05 13:59:30 +08:00
AllentDan 8670d838cb
Fix regression test for mmocr (#2099)
* fix text recog regression test

* update mmocr.yml and change | to + in regression_test.py

* recover
2023-05-22 16:09:11 +08:00
huayuan4396 5e9d27b8d6
mmedit -> mmagic (#2061)
* mmedit -> mmagic --initial

* fix codebase/cmakelist

* add tests/test_codebase/test_mmagic/data/

* fix lint

* fix rename

* fix EditDataPreprocessor

* fix EditTestLoop to TestLoop for mmagic

* fix EditValLoop to ValLoop for mmagic

* fix EditEvaluator to Evaluator for mmagic

* modify rgtest/mmagic.yml

* fix to MultiEvaluator

* fix mmagic model.py

* fix reg_test

* fix lint

* pass rgtest

* fix ci quantize.yml

* fix ci

* update docs

* fix lint

* fix lint

* fix lint

* fix sr end2endmodel device

* change destruct device back to cpu

* modify output device

* rename function name

* update docstring
2023-05-19 15:00:45 +08:00
RunningLeon 1c7749d17c
[Enhancement]: Support opset_version 13 (#2071)
* upgrade to opset 13

* fix unsqueeze

* fix mmseg yml

* fix mmseg reg test

* forcely change opset13

* fix mmdet3d

* optimize squeeze

* update base dockerfile

* support squeeze/unsqueeze with axes as input in onnx2ncnn

* update optimizer for squeeze/unsqueeze

* revert

* Revert "support squeeze/unsqueeze with axes as input in onnx2ncnn"

This reverts commit 5ca9f1ae172cb4e1625f150ccb049138b5f37aa3.

* fix docs

* fix opset
2023-05-17 11:02:30 +08:00
huayuan4396 840adcfb43
mmcls -> mmpretrain (#2003)
* mmcls -> mmpretrain

* add constraints.txt

* fix lint

* fix lint

* remove constraints.txt

* fix windows ci

* modify requirements

* fix mdlink and mmpretrain version

* fix dead link

* modify codebase cmakelist

* fix rename
2023-04-28 20:49:22 +08:00
lihongda 9a442ab4ed
CodeCamp #106: Support SOLOv2 deployment with OpenVINO (#1521)
* support solov2 openvino

* fix lint

* Update benchmark.md

* Update __init__.py

* fix names and delete redundant codes

* fix names and redundant codes

* fix indent

* fix lint

* fix lint

* delete 'ctx' arguments

* [Fix]fix lint

* 修改:     mmdeploy/codebase/mmdet/models/dense_heads/solov2_head.py
	修改:     mmdeploy/codebase/mmdet/models/detectors/single_stage_instance_seg.py

* Delete matrix_nms.py

* Update __init__.py

* Update linspace.py

* Update solov2_head.py

* Update solov2_head.py

* update

* fix num_groups

* update docs

* add end2endmodel for solo

* add end2endmodel for solo

* [Feature] Support End2EndModelSOLO

* Update object_detection_model.py

* add model_cfg to End2EndModel

* add model_cfg to End2EndModel

* Fix bugs for End2EndModel

* fix bugs for End2EndModel

* update docs.

* update conmment

* 更新 benchmark.md

* update regression tests

* fix errors in docs
2023-04-26 11:47:41 +08:00
kaizhong 671d626e33
Generate supported-backends markdown table (#1986)
* Generate supported-backends markdown table

* Generate supported-backends markdown table

* update branch

* update codebase dir

* update backends nargs

* update args to capitals

* center alignment starting from the 3rd coloumn

* center alignment starting from the 3rd coloumn

* fix bad urls

* fix bad urls

* Update codebases.txt

fix circleci

* Update codebases.txt

---------

Co-authored-by: RunningLeon <mnsheng@yeah.net>
2023-04-21 11:12:27 +08:00
AllentDan 6c26e887d4
[Doc] update SAR status (#1789)
* update SAR status and fix torchscript export

* add reminder for SAR
2023-04-12 15:43:51 +08:00
hanrui1sensetime d76c7b61a5
[Fix] Fix errors about deploying MMYOLO-OpenVINO, DETR, ConvFormer and RTMDet (#1919)
* fix reg test yolox

* fix detr

* fix rtmdet-sdk reg

* fix conformer precision

* add conformer_cls sdk

* add mmcls ut

* fix detr ut

* fix detr ut

* fix lint

* fix yapf

* fix cls sdk

* fix detr_head rewriter

* fix interpolate

* complement the mmdet ut

* fix regression DETR"

* fix ut

* fix ut version

* fix lint
2023-03-31 13:45:15 +08:00
RunningLeon 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
Shengxi Li 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
hanrui1sensetime 637958a910
[Fix] Fix MobileOne deployment. (#1870)
* fix task

* fix regression;

* fix ut
2023-03-13 10:54:07 +08:00
AllentDan 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
DDGRCF 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
q.yao 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
AllentDan 3cb4a3fc6a
init (#1423) 2022-11-25 11:55:24 +08:00
RunningLeon 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
RunningLeon 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
hanrui1sensetime 115ca470dd
fix mobileone regtest (#1393) 2022-11-18 16:43:56 +08:00
AllentDan 81eafa5fb3
[Fix] fix regression test for mmocr(#1372)
* fix

* refine
2022-11-18 11:37:26 +08:00
SsTtOoNnEe 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
hanrui1sensetime 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
vansin 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
RunningLeon d330e17af3
fix reg of dev-1.x(#1317) 2022-11-08 11:17:59 +08:00
tpoisonooo 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
RunningLeon 13290614f6
Fix hourglass from mmpose (#1277)
* update mmpose rewritings

* update yml

* update docstring for mmpose
2022-11-03 15:15:33 +08:00
RunningLeon 2fa5095154 cherry-pick from commit 197a7ad 2022-11-01 16:40:12 +08:00
SsTtOoNnEe 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
RunningLeon 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
Chen Xin 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
RunningLeon 6e9c2527b4 merge master from c35099ef0 2022-10-13 11:12:07 +08:00
Qingren 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
RunningLeon 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
RunningLeon 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
RunningLeon 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
hanrui1sensetime 161f27d493
fix mmedit test picture (#1131) 2022-09-29 17:13:23 +08:00
AllentDan 23e84f46ff
update mmcls yaml (#1137) 2022-09-29 17:09:17 +08:00
q.yao d282503012
update detyaml (#1133) 2022-09-29 12:14:06 +08:00
RunningLeon 40adb329a3 Merge master:ea7706cb into sync_master 2022-09-22 19:49:50 +08:00
hanrui1sensetime 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
RunningLeon 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
AllentDan 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
RunningLeon 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
RunningLeon ac3a12026d
fix mmdet ut (#1001) 2022-09-05 19:56:19 +08:00
RunningLeon 966d737a1b
Fix: skip tests for uninstalled codebases (#987)
* skip tests if codebase not installed

* skip ort run test

* fix mmseg
2022-09-03 15:07:08 +08:00
RunningLeon cbedf1cc30
fix regression test (#958)
* fix reg

* set sdk wrapper device id

* resolve comment
2022-09-02 11:28:12 +08:00
RunningLeon 71e64786b7
Fix mmseg pointrend (#903)
* support mmseg:pointrend

* update docs

* update docs for torchscript

* resolve comments
2022-09-02 10:55:32 +08:00
hanrui1sensetime 308e28fcb0
[Enhancement] Support Object Detection and Instance Segmentation for ort trt ncnn and openvino in mmdet 2.0 (#786)
* support cascade (mask) rcnn

* fix docstring

* support SwinTransformer

* move dense_head support to this branch

* fix function names

* fix part of uts of mmdet

* fix for mmdet ut

* fix det model cfg for ut

* fix test_object_detection.py

* fix mmdet object_detection_model.py

* fix mmdet yolov3 ort ut

* fix part of uts

* fix cascade bbox head ut

* fix cascade bbox head ut

* remove useless ssd ncnn test

* fix ncnn wrapper

* fix openvino ut for reppoint head

* fix openvino cascade mask rcnn

* sync codes

* support roll

* remove unused pad

* fix yolox

* fix isort

* fix lint

* fix flake8

* reply for comments and fix failed ut

* fix sdk_export in dump_info

* fix temp hidden xlsx bugs

* fix mmdet regression test

* fix lint

* fix timer

* fix timecount side-effect

* adapt profile.py for mmdet 2.0

* hardcode report.txt for T4 benchmark test: temp version

* fix no-visualizer case

* fix backend_model

* fix android build

* adapt new mmdet 2.0 0825

* fix new 2.0

* fix test_mmdet_structures

* fix test_object_detection

* fix codebase import

* fix ut

* fix all mmdet uts

* fix det

* fix mmdet trt

* fix ncnn onnx optimize
2022-09-01 11:35:57 +08:00
q.yao f4decda86e
[Enhancement] Support DETR (#924)
* add detr support

* fix softmax

* add reg test, update document
2022-08-30 10:39:10 +08:00
AllentDan 0111e1f10f
[Feature] add swin for cls (#911)
* add swin for cls

* add ut and doc

* reduce trt batch size

* add regression test

* resolve comments

* remove useless rewriting logic
2022-08-23 17:14:24 +08:00