Add detection and segmentation examples for Ascend deployment

examples/vision/segmentation/paddleseg/cpp/README.md

@@ -34,6 +34,8 @@ wget https://paddleseg.bj.bcebos.com/dygraph/demo/cityscapes_demo.png
 ./infer_demo Unet_cityscapes_without_argmax_infer cityscapes_demo.png 2
 # 昆仑芯XPU推理
 ./infer_demo Unet_cityscapes_without_argmax_infer cityscapes_demo.png 3
+# 华为昇腾推理
+./infer_demo Unet_cityscapes_without_argmax_infer cityscapes_demo.png 4
 ```
 
 运行完成可视化结果如下图所示

examples/vision/segmentation/paddleseg/cpp/infer.cc

@@ -135,6 +135,34 @@ void TrtInfer(const std::string& model_dir, const std::string& image_file) {
   std::cout << "Visualized result saved in ./vis_result.jpg" << std::endl;
 }
 
+void AscendInfer(const std::string& model_dir, const std::string& image_file) {
+  auto model_file = model_dir + sep + "model.pdmodel";
+  auto params_file = model_dir + sep + "model.pdiparams";
+  auto config_file = model_dir + sep + "deploy.yaml";
+  auto option = fastdeploy::RuntimeOption();
+  option.UseAscend();
+  auto model = fastdeploy::vision::segmentation::PaddleSegModel(
+      model_file, params_file, config_file, option);
+
+  if (!model.Initialized()) {
+    std::cerr << "Failed to initialize." << std::endl;
+    return;
+  }
+
+  auto im = cv::imread(image_file);
+
+  fastdeploy::vision::SegmentationResult res;
+  if (!model.Predict(im, &res)) {
+    std::cerr << "Failed to predict." << std::endl;
+    return;
+  }
+
+  std::cout << res.Str() << std::endl;
+  auto vis_im = fastdeploy::vision::VisSegmentation(im, res, 0.5);
+  cv::imwrite("vis_result.jpg", vis_im);
+  std::cout << "Visualized result saved in ./vis_result.jpg" << std::endl;
+}
+
 int main(int argc, char* argv[]) {
   if (argc < 4) {
     std::cout
@@ -155,6 +183,8 @@ int main(int argc, char* argv[]) {
     TrtInfer(argv[1], argv[2]);
   } else if (std::atoi(argv[3]) == 3) {
     KunlunXinInfer(argv[1], argv[2]);
+  } else if (std::atoi(argv[3]) == 4) {
+    AscendInfer(argv[1], argv[2]);
   }
   return 0;
 }

examples/vision/segmentation/paddleseg/python/README.md

@@ -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
 ```
 
 运行完成可视化结果如下图所示

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],