Add detection and segmentation examples for Ascend deployment

2025-10-16 21:51:31 +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/detection/yolov5/python/infer.py
+++ b/examples/vision/detection/yolov5/python/infer.py
@@ -31,6 +31,9 @@ def build_option(args):
    if args.device.lower() == "gpu":
        option.use_gpu()

+    if args.device.lower() == "ascend":
+        option.use_ascend()
+
    if args.use_trt:
        option.use_trt_backend()
        option.set_trt_input_shape("images", [1, 3, 640, 640])