C# OpenVINO 百度PaddleSeg实时人像抠图PP-MattingV2

目录

效果

项目

代码

下载 

C# OpenVINO 百度PaddleSeg实时人像抠图PP-MattingV2

效果

项目

代码

using OpenCvSharp;
using Sdcb.OpenVINO;
using System;
using System.Diagnostics;
using System.Drawing;
using System.Security.Cryptography;
using System.Text;
using System.Windows.Forms;

namespace C__OpenVINO_Demo
{
    public partial class Form1 : Form
    {
        Bitmap bmp;
        string fileFilter = "*.*|*.bmp;*.jpg;*.jpeg;*.tiff;*.tiff;*.png";
        string img = "";
        string model_path;
        CompiledModel cm;
        InferRequest ir;
        StringBuilder sb = new StringBuilder();

        public Form1()
        {
            InitializeComponent();
        }

        private void Form1_Load(object sender, EventArgs e)
        {
            model_path = "models/ppmattingv2-stdc1-human_512/model.pdmodel";
            Model rawModel = OVCore.Shared.ReadModel(model_path);

            var ad = OVCore.Shared.AvailableDevices;
            Console.WriteLine("可用设备");
            foreach (var item in ad)
            {
                Console.WriteLine(item);
            }

            cm = OVCore.Shared.CompileModel(rawModel, "CPU");
            ir = cm.CreateInferRequest();

            img = "1.jpg";
            bmp = new Bitmap(img);
            pictureBox1.Image = new Bitmap(img);

        }

        private void button1_Click(object sender, EventArgs e)
        {
            OpenFileDialog ofd = new OpenFileDialog();
            ofd.Filter = fileFilter;
            if (ofd.ShowDialog() != DialogResult.OK) return;

            pictureBox1.Image = null;

            img = ofd.FileName;
            bmp = new Bitmap(img);
            pictureBox1.Image = new Bitmap(img);
            textBox1.Text = "";
        }

        private void button2_Click(object sender, EventArgs e)
        {
            if (img == "") { return; }

            textBox1.Text = "";
            sb.Clear();
            pictureBox2.Image = null;
            Application.DoEvents();

            Mat src = new Mat(img);

            Stopwatch stopwatch = new Stopwatch();

            Mat resize_image = new Mat();
            Cv2.Resize(src, resize_image, new OpenCvSharp.Size(512, 512));

            Mat normalized = Common.Normalize(resize_image);

            float[] input_tensor_data = Common.ExtractMat(normalized);

            Tensor input_img = Tensor.FromArray(input_tensor_data, new Shape(1, 3, 512, 512));

            ir.Inputs[0] = input_img;

            double preprocessTime = stopwatch.Elapsed.TotalMilliseconds;
            stopwatch.Restart();

            ir.Run();

            double inferTime = stopwatch.Elapsed.TotalMilliseconds;
            stopwatch.Restart();

            Tensor output_0 = ir.Outputs[0];

            float[] mask = output_0.GetData().ToArray();

            Mat mask_mat = new Mat(512, 512, MatType.CV_32FC1, mask);

            mask_mat *= 255;
            mask_mat.ConvertTo(mask_mat, MatType.CV_8UC1);

            Cv2.Resize(mask_mat, mask_mat, new OpenCvSharp.Size(512, 512));

            double postprocessTime = stopwatch.Elapsed.TotalMilliseconds;
            stopwatch.Stop();
            double totalTime = preprocessTime + inferTime + postprocessTime;

            sb.AppendLine($"Preprocess: {preprocessTime:F2}ms");
            sb.AppendLine($"Infer: {inferTime:F2}ms");
            sb.AppendLine($"Postprocess: {postprocessTime:F2}ms");
            sb.AppendLine($"Total: {totalTime:F2}ms");

            textBox1.Text = sb.ToString();

            Cv2.ImShow("mask", mask_mat);
            //string s = mask_mat.Dump();

            Cv2.Merge(new Mat[] { mask_mat, mask_mat, mask_mat }, mask_mat);

            Mat dst = new Mat();

            Cv2.BitwiseAnd(resize_image, mask_mat, dst);
            //或者 dst = mask_mat & resize_image;
            //string s2 = dst.Dump();

            //透明图处理
            //B,G,R,A   B,G,R -> 0 黑色 255 白色
            //A为透明度 -> 255为不透明，0为全透。
            //Cv2.ImShow("dst", dst);

            var srcBgr = Cv2.Split(dst);
            var alphaBgr = Cv2.Split(mask_mat);
            var bgra = new[] { srcBgr[0], srcBgr[1], srcBgr[2], alphaBgr[0] };
            Cv2.Merge(bgra, dst);

            //Cv2.ImShow("dst_alpha_0", dst);
            Cv2.ImWrite("1.png", dst);
            //string s3 = dst.Dump();

            pictureBox2.Image = new System.Drawing.Bitmap(dst.ToMemoryStream());

        }

    }

}

  1. using OpenCvSharp;
  2. using Sdcb.OpenVINO;
  3. using System;
  4. using System.Diagnostics;
  5. using System.Drawing;
  6. using System.Security.Cryptography;
  7. using System.Text;
  8. using System.Windows.Forms;
  9.  
  10. namespace C__OpenVINO_Demo
  11. {
  12.     public partial class Form1 : Form
  13.     {
  14.         Bitmap bmp;
  15.         string fileFilter = "*.*|*.bmp;*.jpg;*.jpeg;*.tiff;*.tiff;*.png";
  16.         string img = "";
  17.         string model_path;
  18.         CompiledModel cm;
  19.         InferRequest ir;
  20.         StringBuilder sb = new StringBuilder();
  21.  
  22.         public Form1()
  23.         {
  24.             InitializeComponent();
  25.         }
  26.  
  27.         private void Form1_Load(object sender, EventArgs e)
  28.         {
  29.             model_path = "models/ppmattingv2-stdc1-human_512/model.pdmodel";
  30.             Model rawModel = OVCore.Shared.ReadModel(model_path);
  31.  
  32.             var ad = OVCore.Shared.AvailableDevices;
  33.             Console.WriteLine("可用设备");
  34.             foreach (var item in ad)
  35.             {
  36.                 Console.WriteLine(item);
  37.             }
  38.  
  39.             cm = OVCore.Shared.CompileModel(rawModel, "CPU");
  40.             ir = cm.CreateInferRequest();
  41.  
  42.             img = "1.jpg";
  43.             bmp = new Bitmap(img);
  44.             pictureBox1.Image = new Bitmap(img);
  45.  
  46.         }
  47.  
  48.         private void button1_Click(object sender, EventArgs e)
  49.         {
  50.             OpenFileDialog ofd = new OpenFileDialog();
  51.             ofd.Filter = fileFilter;
  52.             if (ofd.ShowDialog() != DialogResult.OK) return;
  53.  
  54.             pictureBox1.Image = null;
  55.  
  56.             img = ofd.FileName;
  57.             bmp = new Bitmap(img);
  58.             pictureBox1.Image = new Bitmap(img);
  59.             textBox1.Text = "";
  60.         }
  61.  
  62.         private void button2_Click(object sender, EventArgs e)
  63.         {
  64.             if (img == "") { return; }
  65.  
  66.             textBox1.Text = "";
  67.             sb.Clear();
  68.             pictureBox2.Image = null;
  69.             Application.DoEvents();
  70.  
  71.             Mat src = new Mat(img);
  72.  
  73.             Stopwatch stopwatch = new Stopwatch();
  74.  
  75.             Mat resize_image = new Mat();
  76.             Cv2.Resize(src, resize_image, new OpenCvSharp.Size(512, 512));
  77.  
  78.             Mat normalized = Common.Normalize(resize_image);
  79.  
  80.             float[] input_tensor_data = Common.ExtractMat(normalized);
  81.  
  82.             Tensor input_img = Tensor.FromArray(input_tensor_data, new Shape(1, 3, 512, 512));
  83.  
  84.             ir.Inputs[0] = input_img;
  85.  
  86.             double preprocessTime = stopwatch.Elapsed.TotalMilliseconds;
  87.             stopwatch.Restart();
  88.  
  89.             ir.Run();
  90.  
  91.             double inferTime = stopwatch.Elapsed.TotalMilliseconds;
  92.             stopwatch.Restart();
  93.  
  94.             Tensor output_0 = ir.Outputs[0];
  95.  
  96.             float[] mask = output_0.GetData<float>().ToArray();
  97.  
  98.             Mat mask_mat = new Mat(512, 512, MatType.CV_32FC1, mask);
  99.  
  100.             mask_mat *= 255;
  101.             mask_mat.ConvertTo(mask_mat, MatType.CV_8UC1);
  102.  
  103.             Cv2.Resize(mask_mat, mask_mat, new OpenCvSharp.Size(512, 512));
  104.  
  105.             double postprocessTime = stopwatch.Elapsed.TotalMilliseconds;
  106.             stopwatch.Stop();
  107.             double totalTime = preprocessTime + inferTime + postprocessTime;
  108.  
  109.             sb.AppendLine($"Preprocess: {preprocessTime:F2}ms");
  110.             sb.AppendLine($"Infer: {inferTime:F2}ms");
  111.             sb.AppendLine($"Postprocess: {postprocessTime:F2}ms");
  112.             sb.AppendLine($"Total: {totalTime:F2}ms");
  113.  
  114.             textBox1.Text = sb.ToString();
  115.  
  116.             Cv2.ImShow("mask", mask_mat);
  117.             //string s = mask_mat.Dump();
  118.  
  119.             Cv2.Merge(new Mat[] { mask_mat, mask_mat, mask_mat }, mask_mat);
  120.  
  121.             Mat dst = new Mat();
  122.  
  123.             Cv2.BitwiseAnd(resize_image, mask_mat, dst);
  124.             //或者 dst = mask_mat & resize_image;
  125.             //string s2 = dst.Dump();
  126.  
  127.             //透明图处理
  128.             //B,G,R,A   B,G,R -> 0 黑色 255 白色
  129.             //A为透明度 -> 255为不透明，0为全透。
  130.             //Cv2.ImShow("dst", dst);
  131.  
  132.             var srcBgr = Cv2.Split(dst);
  133.             var alphaBgr = Cv2.Split(mask_mat);
  134.             var bgra = new[] { srcBgr[0], srcBgr[1], srcBgr[2], alphaBgr[0] };
  135.             Cv2.Merge(bgra, dst);
  136.  
  137.             //Cv2.ImShow("dst_alpha_0", dst);
  138.             Cv2.ImWrite("1.png", dst);
  139.             //string s3 = dst.Dump();
  140.  
  141.             pictureBox2.Image = new System.Drawing.Bitmap(dst.ToMemoryStream());
  142.  
  143.         }
  144.  
  145.     }
  146.  
  147. }

下载 

源码下载

天天代码码天天
微信公众号
.NET 人工智能实践
注：本文转载自blog.csdn.net的天天代码码天天的文章"https://lw112190.blog.csdn.net/article/details/136260539"。版权归原作者所有，此博客不拥有其著作权，亦不承担相应法律责任。如有侵权，请联系我们删除。
复制链接
相关推荐
发表评论

评论记录：

未查询到任何数据！