Add detection and segmentation examples for Ascend deployment

2025-10-16 05:30:58 +08:00 · 2022-12-27 07:40:46 +00:00
parent 76a876406e
commit 090c3a68b4
27 changed files with 280 additions and 26 deletions
--- a/examples/vision/segmentation/paddleseg/python/infer.py
+++ b/examples/vision/segmentation/paddleseg/python/infer.py
@@ -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],