mirrors
/
mmdeploy
mirror of https://github.com/open-mmlab/mmdeploy.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix suffix (#1345)

pull/1361/head
AllentDan 2022-11-10 16:15:17 +08:00 committed by GitHub
parent ccc21289d1
commit 3eb53ad891
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mmdeploy/backend/sdk/export_info.py
View File

@ -90,8 +90,8 @@ def get_models(deploy_cfg: Union[str, mmengine.Config],
tensorrt=lambda file: re.sub(r'\.[a-z]+', '.engine', file), tensorrt=lambda file: re.sub(r'\.[a-z]+', '.engine', file),
openvino=lambda file: re.sub(r'\.[a-z]+', '.xml', file), openvino=lambda file: re.sub(r'\.[a-z]+', '.xml', file),
ncnn=lambda file: re.sub(r'\.[a-z]+', '.param', file), ncnn=lambda file: re.sub(r'\.[a-z]+', '.param', file),
ascend=lambda file: re.sub(r'\.[a-z]+', '.rknn', file), ascend=lambda file: re.sub(r'\.[a-z]+', '.om', file),
rknn=lambda file: re.sub(r'\.[a-z]+', '.om', file), rknn=lambda file: re.sub(r'\.[a-z]+', '.rknn', file),
coreml=lambda file: re.sub(r'\.[a-z]+', '.mlpackage', file), coreml=lambda file: re.sub(r'\.[a-z]+', '.mlpackage', file),
snpe=lambda file: re.sub(r'\.[a-z]+', '.dlc', file)) snpe=lambda file: re.sub(r'\.[a-z]+', '.dlc', file))
backend_weights = dict( backend_weights = dict(