mmdeploy/configs/mmseg/segmentation_coreml_static-512x1024.py
Chen Xin a0fb3be0df
support coreml (#760)
* sdk inference

* fix typo

* fix typo

* add convert things

* fix missling name

* add cls support

* add more pytorch rewriter

* add det support

* support det wip

* make Model export model_path

* fix nms

* add output back

* add docstring

* fix lint

* add coreml build action

* add zh docs

* add coreml backend check

* update ci

* update

* update

* update

* update

* update

* fix lint

* update configs

* add return value when error occured

* update docs

* update docs

* update docs

* fix lint

* udpate docs

* udpate docs

* update

Co-authored-by: grimoire <streetyao@live.com>
2022-09-05 19:55:47 +08:00

15 lines
400 B
Python

_base_ = [
'../_base_/torchscript_config.py', '../_base_/backends/coreml.py',
'./segmentation_static.py'
]
ir_config = dict(input_shape=[1024, 512])
backend_config = dict(model_inputs=[
dict(
input_shapes=dict(
input=dict(
min_shape=[1, 3, 512, 1024],
max_shape=[1, 3, 512, 1024],
default_shape=[1, 3, 512, 1024])))
])