mmdeploy/classification_ascend_static-224x224.py at 6775d031b3d9cc77c228a217bfd16d193cce0837 - 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

Li Zhang 792c27b054

[Feature] Ascend backend (#747 )

* add acl backend

* support dynamic batch size and dynamic image size

* add preliminary ascend backend

* support dtypes other than float

* support dynamic_dims in SDK

* fix dynamic batch size

* better error handling

* remove debug info

* [WIP] dynamic shape support

* fix static shape

* fix dynamic batch size

* add retinanet support

* fix dynamic image size

* fix dynamic image size

* fix dynamic dims

* fix dynamic dims

* simplify config files

* fix yolox support

* fix negative index

* support faster rcnn

* add seg config

* update benchmark

* fix onnx2ascend dynamic shape

* update docstring and benchmark

* add unit test, update documents

* fix wrapper

* fix ut

* fix for vit

* error handling

* context handling & multi-device support

* build with stub libraries

* add ci

* fix lint

* fix lint

* update doc ref

* fix typo

* down with `target_link_directories`

* setup python

* makedir

* fix ci

* fix ci

* remove verbose logs

* fix UBs

* export Error

* fix lint

* update checkenv

Co-authored-by: grimoire <yaoqian@sensetime.com>

2022-09-05 12:08:36 +08:00

6 lines

207 B

Python

Raw Blame History

 _base_ = ['./classification_static.py', '../_base_/backends/ascend.py']
 onnx_config = dict(input_shape=[224, 224])
 backend_config = dict(
     model_inputs=[dict(input_shapes=dict(input=[1, 3, 224, 224]))])