mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
124635ec5f
* save codes * support resnet and yolov3 * support yolox * fix lint * add mmseg support and a doc * add UT * update supported model list * fix ci * refine docstring * resolve comments * remote output_tensor_type * resolve comments * update readme
12 lines
350 B
Python
12 lines
350 B
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
from mmdeploy.backend.rknn import is_available
|
|
|
|
__all__ = ['is_available']
|
|
|
|
if is_available():
|
|
from mmdeploy.backend.rknn.onnx2rknn import onnx2rknn as _onnx2rknn
|
|
from ..core import PIPELINE_MANAGER
|
|
onnx2rknn = PIPELINE_MANAGER.register_pipeline()(_onnx2rknn)
|
|
|
|
__all__ += ['onnx2rknn']
|