C# OpenVINO 图片旋转角度检测

目录

效果

项目

代码

下载 

效果

项目

代码

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

namespace C__OpenVINO_图片旋转角度检测
{
    public partial class Form1 : Form
    {
        Bitmap bmp;
        string fileFilter = "*.*|*.bmp;*.jpg;*.jpeg;*.tiff;*.tiff;*.png";
        string img = "";
        float rotateThreshold = 0.50f;
        InputShape defaultShape = new InputShape(3, 224, 224);
        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/inference.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 button3_Click(object sender, EventArgs e)
        {
            if (bmp == null)
            {
                return;
            }

            var mat = OpenCvSharp.Extensions.BitmapConverter.ToMat(bmp);
            Cv2.CvtColor(mat, mat, ColorConversionCodes.RGBA2RGB);
            Cv2.Rotate(mat, mat, RotateFlags.Rotate90Clockwise);
            var bitmap = OpenCvSharp.Extensions.BitmapConverter.ToBitmap(mat);
            pictureBox1.Image = bitmap;
        }

        private void button4_Click(object sender, EventArgs e)
        {
            if (bmp == null)
            {
                return;
            }

            var mat = OpenCvSharp.Extensions.BitmapConverter.ToMat(bmp);
            Cv2.CvtColor(mat, mat, ColorConversionCodes.RGBA2RGB);
            Cv2.Rotate(mat, mat, RotateFlags.Rotate180);
            var bitmap = OpenCvSharp.Extensions.BitmapConverter.ToBitmap(mat);
            pictureBox1.Image = bitmap;
        }

        private void button5_Click(object sender, EventArgs e)
        {
            if (bmp == null)
            {
                return;
            }

            var mat = OpenCvSharp.Extensions.BitmapConverter.ToMat(bmp);
            Cv2.CvtColor(mat, mat, ColorConversionCodes.RGBA2RGB);
            Cv2.Rotate(mat, mat, RotateFlags.Rotate90Counterclockwise);
            var bitmap = OpenCvSharp.Extensions.BitmapConverter.ToBitmap(mat);
            pictureBox1.Image = bitmap;
        }

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

            textBox1.Text = "";
            sb.Clear();

            Mat src = OpenCvSharp.Extensions.BitmapConverter.ToMat(new Bitmap(pictureBox1.Image));
            Cv2.CvtColor(src, src, ColorConversionCodes.RGBA2RGB);//mat转三通道mat

            Stopwatch stopwatch = new Stopwatch();

            Mat resized = Common.ResizePadding(src, defaultShape);
            Mat normalized = Common.Normalize(resized);

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

            Tensor input_x = Tensor.FromArray(input_tensor_data, new Shape(1, 3, 224, 224));

            ir.Inputs[0] = input_x;

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

            ir.Run();

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

            Tensor output_0 = ir.Outputs[0];

            RotationDegree r = RotationDegree._0;

            float[] softmax = output_0.GetData().ToArray();
            float max = softmax.Max();
            int maxIndex = Array.IndexOf(softmax, max);

            if (max > rotateThreshold)
            {
                r = (RotationDegree)maxIndex;
            }

            string result = r.ToString();

            result = result + " (" + max.ToString("P2")+")";

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

            sb.AppendLine("结果：" + result);
            sb.AppendLine();
            sb.AppendLine("Scores: [" + String.Join(", ", softmax) + "]");
            sb.AppendLine();
            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();

        }

    }

    public readonly struct InputShape
    {
        ///


        /// Initializes a new instance of the struct.
        ///
        /// The number of color channels in the input image.
        /// The width of the input image in pixels.
        /// The height of the input image in pixels.
        public InputShape(int channel, int width, int height)
        {
            Channel = channel;
            Height = height;
            Width = width;
        }

        ///


        /// Gets the number of color channels in the input image.
        ///
        public int Channel { get; }

        ///


        /// Gets the height of the input image in pixels.
        ///
        public int Height { get; }

        ///


        /// Gets the width of the input image in pixels.
        ///
        public int Width { get; }
    }

    ///


    /// Enum representing the degrees of rotation.
    ///
    public enum RotationDegree
    {
        ///
        /// Represents the 0-degree rotation angle.
        ///
        _0,
        ///
        /// Represents the 90-degree rotation angle.
        ///
        _90,
        ///
        /// Represents the 180-degree rotation angle.
        ///
        _180,
        ///
        /// Represents the 270-degree rotation angle.
        ///
        _270,
    }
}

  1. using OpenCvSharp;
  2. using Sdcb.OpenVINO;
  3. using System;
  4. using System.Diagnostics;
  5. using System.Drawing;
  6. using System.Linq;
  7. using System.Runtime.InteropServices;
  8. using System.Security.Cryptography;
  9. using System.Text;
  10. using System.Windows.Forms;
  11.  
  12. namespace C__OpenVINO_图片旋转角度检测
  13. {
  14.     public partial class Form1 : Form
  15.     {
  16.         Bitmap bmp;
  17.         string fileFilter = "*.*|*.bmp;*.jpg;*.jpeg;*.tiff;*.tiff;*.png";
  18.         string img = "";
  19.         float rotateThreshold = 0.50f;
  20.         InputShape defaultShape = new InputShape(3, 224, 224);
  21.         string model_path;
  22.         CompiledModel cm;
  23.         InferRequest ir;
  24.  
  25.         StringBuilder sb = new StringBuilder();
  26.  
  27.         public Form1()
  28.         {
  29.             InitializeComponent();
  30.         }
  31.  
  32.         private void Form1_Load(object sender, EventArgs e)
  33.         {
  34.             model_path = "models/inference.pdmodel";
  35.             Model rawModel = OVCore.Shared.ReadModel(model_path);
  36.  
  37.             var ad = OVCore.Shared.AvailableDevices;
  38.             Console.WriteLine("可用设备");
  39.             foreach (var item in ad)
  40.             {
  41.                 Console.WriteLine(item);
  42.             }
  43.  
  44.             cm = OVCore.Shared.CompileModel(rawModel, "CPU");
  45.             ir = cm.CreateInferRequest();
  46.  
  47.             img = "1.jpg";
  48.             bmp = new Bitmap(img);
  49.             pictureBox1.Image = new Bitmap(img);
  50.  
  51.         }
  52.  
  53.         private void button1_Click(object sender, EventArgs e)
  54.         {
  55.             OpenFileDialog ofd = new OpenFileDialog();
  56.             ofd.Filter = fileFilter;
  57.             if (ofd.ShowDialog() != DialogResult.OK) return;
  58.  
  59.             pictureBox1.Image = null;
  60.  
  61.             img = ofd.FileName;
  62.             bmp = new Bitmap(img);
  63.             pictureBox1.Image = new Bitmap(img);
  64.             textBox1.Text = "";
  65.         }
  66.  
  67.         private void button3_Click(object sender, EventArgs e)
  68.         {
  69.             if (bmp == null)
  70.             {
  71.                 return;
  72.             }
  73.  
  74.             var mat = OpenCvSharp.Extensions.BitmapConverter.ToMat(bmp);
  75.             Cv2.CvtColor(mat, mat, ColorConversionCodes.RGBA2RGB);
  76.             Cv2.Rotate(mat, mat, RotateFlags.Rotate90Clockwise);
  77.             var bitmap = OpenCvSharp.Extensions.BitmapConverter.ToBitmap(mat);
  78.             pictureBox1.Image = bitmap;
  79.         }
  80.  
  81.         private void button4_Click(object sender, EventArgs e)
  82.         {
  83.             if (bmp == null)
  84.             {
  85.                 return;
  86.             }
  87.  
  88.             var mat = OpenCvSharp.Extensions.BitmapConverter.ToMat(bmp);
  89.             Cv2.CvtColor(mat, mat, ColorConversionCodes.RGBA2RGB);
  90.             Cv2.Rotate(mat, mat, RotateFlags.Rotate180);
  91.             var bitmap = OpenCvSharp.Extensions.BitmapConverter.ToBitmap(mat);
  92.             pictureBox1.Image = bitmap;
  93.         }
  94.  
  95.         private void button5_Click(object sender, EventArgs e)
  96.         {
  97.             if (bmp == null)
  98.             {
  99.                 return;
  100.             }
  101.  
  102.             var mat = OpenCvSharp.Extensions.BitmapConverter.ToMat(bmp);
  103.             Cv2.CvtColor(mat, mat, ColorConversionCodes.RGBA2RGB);
  104.             Cv2.Rotate(mat, mat, RotateFlags.Rotate90Counterclockwise);
  105.             var bitmap = OpenCvSharp.Extensions.BitmapConverter.ToBitmap(mat);
  106.             pictureBox1.Image = bitmap;
  107.         }
  108.  
  109.         private void button2_Click(object sender, EventArgs e)
  110.         {
  111.             if (img == "") { return; }
  112.  
  113.             textBox1.Text = "";
  114.             sb.Clear();
  115.  
  116.             Mat src = OpenCvSharp.Extensions.BitmapConverter.ToMat(new Bitmap(pictureBox1.Image));
  117.             Cv2.CvtColor(src, src, ColorConversionCodes.RGBA2RGB);//mat转三通道mat
  118.  
  119.             Stopwatch stopwatch = new Stopwatch();
  120.  
  121.             Mat resized = Common.ResizePadding(src, defaultShape);
  122.             Mat normalized = Common.Normalize(resized);
  123.  
  124.             float[] input_tensor_data = Common.ExtractMat(normalized);
  125.  
  126.             Tensor input_x = Tensor.FromArray(input_tensor_data, new Shape(1, 3, 224, 224));
  127.  
  128.             ir.Inputs[0] = input_x;
  129.  
  130.             double preprocessTime = stopwatch.Elapsed.TotalMilliseconds;
  131.             stopwatch.Restart();
  132.  
  133.             ir.Run();
  134.  
  135.             double inferTime = stopwatch.Elapsed.TotalMilliseconds;
  136.             stopwatch.Restart();
  137.  
  138.             Tensor output_0 = ir.Outputs[0];
  139.  
  140.             RotationDegree r = RotationDegree._0;
  141.  
  142.             float[] softmax = output_0.GetData<float>().ToArray();
  143.             float max = softmax.Max();
  144.             int maxIndex = Array.IndexOf(softmax, max);
  145.  
  146.             if (max > rotateThreshold)
  147.             {
  148.                 r = (RotationDegree)maxIndex;
  149.             }
  150.  
  151.             string result = r.ToString();
  152.  
  153.             result = result + " (" + max.ToString("P2")+")";
  154.  
  155.             double postprocessTime = stopwatch.Elapsed.TotalMilliseconds;
  156.             stopwatch.Stop();
  157.             double totalTime = preprocessTime + inferTime + postprocessTime;
  158.  
  159.             sb.AppendLine("结果：" + result);
  160.             sb.AppendLine();
  161.             sb.AppendLine("Scores: [" + String.Join(", ", softmax) + "]");
  162.             sb.AppendLine();
  163.             sb.AppendLine($"Preprocess: {preprocessTime:F2}ms");
  164.             sb.AppendLine($"Infer: {inferTime:F2}ms");
  165.             sb.AppendLine($"Postprocess: {postprocessTime:F2}ms");
  166.             sb.AppendLine($"Total: {totalTime:F2}ms");
  167.  
  168.             textBox1.Text = sb.ToString();
  169.  
  170.         }
  171.  
  172.     }
  173.  
  174.     public readonly struct InputShape
  175.     {
  176.         /// <summary>
  177.         /// Initializes a new instance of the <see cref="InputShape"/> struct.
  178.         /// </summary>
  179.         /// <param name="channel">The number of color channels in the input image.</param>
  180.         /// <param name="width">The width of the input image in pixels.</param>
  181.         /// <param name="height">The height of the input image in pixels.</param>
  182.         public InputShape(int channel, int width, int height)
  183.         {
  184.             Channel = channel;
  185.             Height = height;
  186.             Width = width;
  187.         }
  188.  
  189.         /// <summary>
  190.         /// Gets the number of color channels in the input image.
  191.         /// </summary>
  192.         public int Channel { get; }
  193.  
  194.         /// <summary>
  195.         /// Gets the height of the input image in pixels.
  196.         /// </summary>
  197.         public int Height { get; }
  198.  
  199.         /// <summary>
  200.         /// Gets the width of the input image in pixels.
  201.         /// </summary>
  202.         public int Width { get; }
  203.     }
  204.  
  205.     /// <summary>
  206.     /// Enum representing the degrees of rotation.
  207.     /// </summary>
  208.     public enum RotationDegree
  209.     {
  210.         /// <summary>
  211.         /// Represents the 0-degree rotation angle.
  212.         /// </summary>
  213.         _0,
  214.         /// <summary>
  215.         /// Represents the 90-degree rotation angle.
  216.         /// </summary>
  217.         _90,
  218.         /// <summary>
  219.         /// Represents the 180-degree rotation angle.
  220.         /// </summary>
  221.         _180,
  222.         /// <summary>
  223.         /// Represents the 270-degree rotation angle.
  224.         /// </summary>
  225.         _270,
  226.     }
  227. }

下载 

源码下载

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

评论记录：

未查询到任何数据！