mmdeploy/configs/mmcls/classification_coreml_dynamic-224x224-224x224.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

13 lines
378 B
Python

_base_ = ['../_base_/torchscript_config.py', '../_base_/backends/coreml.py']
codebase_config = dict(type='mmcls', task='Classification')
backend_config = dict(model_inputs=[
dict(
input_shapes=dict(
input=dict(
min_shape=[1, 3, 224, 224],
max_shape=[8, 3, 224, 224],
default_shape=[1, 3, 224, 224])))
])