mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
2968a18d80
* fix(tools/deploy.py): support snpe * improvement(backend/snpe): review advices * docs(backend/snpe): update build * docs(backend/snpe): server support specify port * docs(backend/snpe): update path * fix(backend/snpe): time counter missing argument * docs(backend/snpe): add missing argument * docs(backend/snpe): update download and using * improvement(snpe_net.cpp): load model with modeldata
38 lines
956 B
Python
38 lines
956 B
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
import argparse
|
|
import logging
|
|
|
|
from mmdeploy.apis.snpe import from_onnx
|
|
from mmdeploy.utils import get_root_logger
|
|
|
|
|
|
def parse_args():
|
|
parser = argparse.ArgumentParser(
|
|
description='Convert ONNX to snpe dlc format.')
|
|
parser.add_argument('onnx_path', help='ONNX model path')
|
|
parser.add_argument('output_prefix', help='output snpe dlc model path')
|
|
parser.add_argument(
|
|
'--log-level',
|
|
help='set log level',
|
|
default='INFO',
|
|
choices=list(logging._nameToLevel.keys()))
|
|
args = parser.parse_args()
|
|
|
|
return args
|
|
|
|
|
|
def main():
|
|
args = parse_args()
|
|
logger = get_root_logger(log_level=args.log_level)
|
|
|
|
onnx_path = args.onnx_path
|
|
output_prefix = args.output_prefix
|
|
|
|
logger.info(f'onnx2dlc: \n\tonnx_path: {onnx_path} ')
|
|
from_onnx(onnx_path, output_prefix)
|
|
logger.info('onnx2dlc success.')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|