apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-16 21:51:31 +08:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

Add detection and segmentation examples for Ascend deployment

Browse Source
This commit is contained in:
yunyaoXYY
2022-12-27 07:40:46 +00:00
parent 76a876406e
commit 090c3a68b4
27 changed files with 280 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/vision/segmentation/paddleseg/python/README.md
View File

@@ -27,6 +27,8 @@ python infer.py --model Unet_cityscapes_without_argmax_infer --image cityscapes_
python infer.py --model Unet_cityscapes_without_argmax_infer --image cityscapes_demo.png --device gpu --use_trt True
# 昆仑芯XPU推理
python infer.py --model Unet_cityscapes_without_argmax_infer --image cityscapes_demo.png --device kunlunxin
# 华为昇腾推理
python infer.py --model Unet_cityscapes_without_argmax_infer --image cityscapes_demo.png --device ascend
```
运行完成可视化结果如下图所示

3
examples/vision/segmentation/paddleseg/python/infer.py
View File

@@ -33,6 +33,9 @@ def build_option(args):
if args.device.lower() == "kunlunxin":
option.use_kunlunxin()
if args.device.lower() == "ascend":
option.use_ascend()
if args.use_trt:
option.use_trt_backend()
option.set_trt_input_shape("x", [1, 3, 256, 256], [1, 3, 1024, 1024],