mmdeploy/segmentation_sdk_dynamic.py at 97e0d1228fc5788bceea70a8dbe9b3b4818b230d - mmdeploy - Gitea: Git with a cup of tea

mirrors/mmdeploy

mirror of https://github.com/open-mmlab/mmdeploy.git synced 2025-01-14 08:09:43 +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

11 lines

307 B

Python

Raw Blame History

 _base_ = ['./segmentation_dynamic.py', '../_base_/backends/sdk.py']
 codebase_config = dict(model_type='sdk')
 backend_config = dict(pipeline=[
     dict(type='LoadImageFromFile'),
     dict(type='LoadAnnotations'),
     dict(
         type='PackSegInputs', meta_keys=['img_path', 'ori_shape', 'img_shape'])
 ])