apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-05 16:48:03 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

[Docs] Pick paddleclas fastdeploy docs from PaddleClas (#1654)

Browse Source 
* Adjust folders structures in paddleclas

* remove useless files

* Update sophgo

* improve readme
This commit is contained in:
yunyaoXYY
2023-03-23 13:06:09 +08:00
committed by GitHub
parent ab65557121
commit c91e99b5f5
90 changed files with 2005 additions and 2584 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
examples/vision/classification/paddleclas/sophgo/python/infer.py Executable file → Normal file
View File

@@ -3,13 +3,18 @@ import cv2
import os
from subprocess import run
def parse_arguments():
import argparse
import ast
parser = argparse.ArgumentParser()
parser.add_argument("--auto", required=True, help="Auto download, convert, compile and infer if True")
parser.add_argument(
"--auto",
required=True,
help="Auto download, convert, compile and infer if True")
parser.add_argument("--model", required=True, help="Path of bmodel")
parser.add_argument("--config_file", required=True, help="Path of config file")
parser.add_argument(
"--config_file", required=True, help="Path of config file")
parser.add_argument(
"--image", type=str, required=True, help="Path of test image file.")
parser.add_argument(
@@ -17,6 +22,7 @@ def parse_arguments():
return parser.parse_args()
def download():
cmd_str = 'wget https://bj.bcebos.com/paddlehub/fastdeploy/ResNet50_vd_infer.tgz'
jpg_str = 'wget https://gitee.com/paddlepaddle/PaddleClas/raw/release/2.4/deploy/images/ImageNet/ILSVRC2012_val_00000010.jpeg'
@@ -27,27 +33,33 @@ def download():
run(jpg_str, shell=True)
run(tar_str, shell=True)
def paddle2onnx():
cmd_str = 'paddle2onnx --model_dir ResNet50_vd_infer \
--model_filename inference.pdmodel \
--params_filename inference.pdiparams \
--save_file ResNet50_vd_infer.onnx \
--enable_dev_version True'
print(cmd_str)
run(cmd_str, shell=True)
def mlir_prepare():
mlir_path = os.getenv("MODEL_ZOO_PATH")
mlir_path = mlir_path[:-13]
cmd_list = ['mkdir ResNet50',
'cp -rf ' + os.path.join(mlir_path, 'regression/dataset/COCO2017/') + ' ./ResNet50',
'cp -rf ' + os.path.join(mlir_path, 'regression/image/') + ' ./ResNet50',
'cp ResNet50_vd_infer.onnx ./ResNet50/',
'mkdir ./ResNet50/workspace']
cmd_list = [
'mkdir ResNet50', 'cp -rf ' + os.path.join(
mlir_path, 'regression/dataset/COCO2017/') + ' ./ResNet50',
'cp -rf ' + os.path.join(mlir_path,
'regression/image/') + ' ./ResNet50',
'cp ResNet50_vd_infer.onnx ./ResNet50/', 'mkdir ./ResNet50/workspace'
]
for str in cmd_list:
print(str)
run(str, shell=True)
def onnx2mlir():
cmd_str = 'model_transform.py \
--model_name ResNet50_vd_infer \
@@ -61,11 +73,13 @@ def onnx2mlir():
--test_input ../image/dog.jpg \
--test_result ./ResNet50_vd_infer_top_outputs.npz \
--mlir ./ResNet50_vd_infer.mlir'
print(cmd_str)
os.chdir('./ResNet50/workspace/')
run(cmd_str, shell=True)
os.chdir('../../')
def mlir2bmodel():
cmd_str = 'model_deploy.py \
--mlir ./ResNet50_vd_infer.mlir \
@@ -74,6 +88,7 @@ def mlir2bmodel():
--test_input ./ResNet50_vd_infer_in_f32.npz \
--test_reference ./ResNet50_vd_infer_top_outputs.npz \
--model ./ResNet50_vd_infer_1684x_f32.bmodel'
print(cmd_str)
os.chdir('./ResNet50/workspace')
run(cmd_str, shell=True)
@@ -82,7 +97,7 @@ def mlir2bmodel():
args = parse_arguments()
if(args.auto):
if (args.auto):
download()
paddle2onnx()
mlir_prepare()