devbox
小小林熬夜学编程
这个屌丝很懒，什么也没留下！
关注作者
热门标签
热门文章
当前位置:   article > 正文

Paddle OCR V4 测试Demo_ppocrv4

作者：小小林熬夜学编程 | 2024-06-06 01:57:00

赞

踩

ppocrv4

目录

效果

项目

代码

下载 

效果

项目

VS2022+.net4.8+OCRV4

代码

using OpenCvSharp;
using Sdcb.PaddleInference;
using Sdcb.PaddleOCR;
using Sdcb.PaddleOCR.Models;
using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Data;
using System.Drawing;
using System.Linq;
using System.Text;
using System.Windows.Forms;

namespace PaddleInference.OCRServiceV4
{
    public partial class frmTest : Form
    {
        public frmTest()
        {
            InitializeComponent();
        }

        String startupPath;
        public PaddleOcrAll paddleOcr;
        private string fileFilter = "*.*|*.bmp;*.jpg;*.jpeg;*.tiff;*.tiff;*.png";
        Bitmap bmp;
        PaddleOcrResult ocrResult;
        DateTime dt1 = DateTime.Now;
        DateTime dt2 = DateTime.Now;
        String imgPath;
        Mat mat;
        Pen pen = new Pen(Brushes.Red, 2);

        private void frmTest_Load(object sender, EventArgs e)
        {
            comboBox1.SelectedIndex = 0;
            startupPath = Application.StartupPath;
            rbOnnx.Checked = true;
        }

        private void btnLoadModel_Click(object sender, EventArgs e)
        {
            LoadModel();
            MessageBox.Show("加载完成");
        }

        void LoadModel()
        {
            string detectionModelDir;
            string classificationModelDir;
            string recognitionModelDir;
            string labelFilePath;

            if (comboBox1.SelectedIndex == 0)
            {
                detectionModelDir = startupPath + "\\inference\\ch_PP-OCRv4_det_infer";
                classificationModelDir = startupPath + "\\inference\\ch_ppocr_mobile_v2.0_cls_infer";
                recognitionModelDir = startupPath + "\\inference\\ch_PP-OCRv4_rec_infer";
                labelFilePath = startupPath + "\\inference\\ppocr_keys.txt";
            }
            else
            {
                detectionModelDir = startupPath + "\\inference_server\\detv4_teacher_inference";
                classificationModelDir = startupPath + "\\inference_server\\ch_ppocr_mobile_v2.0_cls_infer";
                recognitionModelDir = startupPath + "\\inference_server\\ch_PP-OCRv4_rec_server_infer";
                labelFilePath = startupPath + "\\inference_server\\ppocr_keys.txt";
            }

            FullOcrModel model = FullOcrModel.FromDirectory(detectionModelDir, classificationModelDir, recognitionModelDir, labelFilePath, ModelVersion.V4);

            if (rbOnnx.Enabled == true)
            {
                paddleOcr = new PaddleOcrAll(model, PaddleDevice.Onnx());
            }

            else if (rbOpenblas.Enabled == true)
            {
                paddleOcr = new PaddleOcrAll(model, PaddleDevice.Openblas());
            }

            else if (rbMkldnn.Enabled == true)
            {
                paddleOcr = new PaddleOcrAll(model, PaddleDevice.Mkldnn());
            }
            else
            {
                paddleOcr = new PaddleOcrAll(model);
            }

            paddleOcr.AllowRotateDetection = true;    /* 允许识别有角度的文字 */
            paddleOcr.Enable180Classification = false; /* 允许识别旋转角度大于90度的文字 */
        }


        private void btnSelect_Click(object sender, EventArgs e)
        {
            if (paddleOcr == null)
            {
                MessageBox.Show("请先加载模型");
                return;
            }


            OpenFileDialog ofd = new OpenFileDialog();
            ofd.Filter = fileFilter;
            if (ofd.ShowDialog() != DialogResult.OK) return;

            imgPath = ofd.FileName;
            bmp = new Bitmap(imgPath, true);

            RotateImage(bmp);

            pictureBox1.Image = bmp;

            richTextBox1.Clear();

            mat = new Mat(ofd.FileName);
            //pictureBox1.Image = OpenCvSharp.Extensions.BitmapConverter.ToBitmap(mat);


            dt1 = DateTime.Now;
            ocrResult = paddleOcr.Run(mat);
            dt2 = DateTime.Now;


            ShowOCRResult(ocrResult);

        }

        /// <summary>
        /// 显示结果
        /// </summary>
        private void ShowOCRResult(PaddleOcrResult ocrResult)
        {

            richTextBox1.Clear();
            Bitmap bitmap = (Bitmap)this.pictureBox1.Image;


            richTextBox1.AppendText("耗时:" + (dt2 - dt1).TotalMilliseconds + "ms\n");
            richTextBox1.AppendText("-----------------------------------\n");
            richTextBox1.AppendText(ocrResult.Text + "\n");

            using (Graphics g = Graphics.FromImage(bitmap))
            {
                foreach (var item in ocrResult.Regions)
                {
                    g.DrawRectangle(pen, item.Rect.BoundingRect().Left, item.Rect.BoundingRect().Top, item.Rect.BoundingRect().Width, item.Rect.BoundingRect().Height);
                }
            }

            pictureBox1.Image = null;
            pictureBox1.Image = bitmap;
        }

        private void button1_Click(object sender, EventArgs e)
        {
            if (paddleOcr == null)
            {
                MessageBox.Show("请先加载模型");
                return;
            }

            if (pictureBox1.Image == null)
            {
                return;
            }
            richTextBox1.Clear();
            Application.DoEvents();

            dt1 = DateTime.Now;
            ocrResult = paddleOcr.Run(mat);
            dt2 = DateTime.Now;

            ShowOCRResult(ocrResult);
        }

        /// <summary>
        /// 根据图片exif调整方向
        /// </summary>
        /// <param name="img"></param>
        public void RotateImage(Bitmap img)
        {
            var exif = img.PropertyItems;
            byte orien = 0;
            var item = exif.Where(m => m.Id == 274).ToArray();
            if (item.Length > 0)
                orien = item[0].Value[0];
            switch (orien)
            {
                case 2:
                    img.RotateFlip(RotateFlipType.RotateNoneFlipX);//horizontal flip
                    break;
                case 3:
                    img.RotateFlip(RotateFlipType.Rotate180FlipNone);//right-top
                    break;
                case 4:
                    img.RotateFlip(RotateFlipType.RotateNoneFlipY);//vertical flip
                    break;
                case 5:
                    img.RotateFlip(RotateFlipType.Rotate90FlipX);
                    break;
                case 6:
                    img.RotateFlip(RotateFlipType.Rotate90FlipNone);//right-top
                    break;
                case 7:
                    img.RotateFlip(RotateFlipType.Rotate270FlipX);
                    break;
                case 8:
                    img.RotateFlip(RotateFlipType.Rotate270FlipNone);//left-bottom
                    break;
                default:
                    break;
            }
        }

    }
}
 

  1. using OpenCvSharp;
  2. using Sdcb.PaddleInference;
  3. using Sdcb.PaddleOCR;
  4. using Sdcb.PaddleOCR.Models;
  5. using System;
  6. using System.Collections.Generic;
  7. using System.ComponentModel;
  8. using System.Data;
  9. using System.Drawing;
  10. using System.Linq;
  11. using System.Text;
  12. using System.Windows.Forms;
  13.  
  14. namespace PaddleInference.OCRServiceV4
  15. {
  16.     public partial class frmTest : Form
  17.     {
  18.         public frmTest()
  19.         {
  20.             InitializeComponent();
  21.         }
  22.  
  23.         String startupPath;
  24.         public PaddleOcrAll paddleOcr;
  25.         private string fileFilter = "*.*|*.bmp;*.jpg;*.jpeg;*.tiff;*.tiff;*.png";
  26.         Bitmap bmp;
  27.         PaddleOcrResult ocrResult;
  28.         DateTime dt1 = DateTime.Now;
  29.         DateTime dt2 = DateTime.Now;
  30.         String imgPath;
  31.         Mat mat;
  32.         Pen pen = new Pen(Brushes.Red, 2);
  33.  
  34.         private void frmTest_Load(object sender, EventArgs e)
  35.         {
  36.             comboBox1.SelectedIndex = 0;
  37.             startupPath = Application.StartupPath;
  38.             rbOnnx.Checked = true;
  39.         }
  40.  
  41.         private void btnLoadModel_Click(object sender, EventArgs e)
  42.         {
  43.             LoadModel();
  44.             MessageBox.Show("加载完成");
  45.         }
  46.  
  47.         void LoadModel()
  48.         {
  49.             string detectionModelDir;
  50.             string classificationModelDir;
  51.             string recognitionModelDir;
  52.             string labelFilePath;
  53.  
  54.             if (comboBox1.SelectedIndex == 0)
  55.             {
  56.                 detectionModelDir = startupPath + "\\inference\\ch_PP-OCRv4_det_infer";
  57.                 classificationModelDir = startupPath + "\\inference\\ch_ppocr_mobile_v2.0_cls_infer";
  58.                 recognitionModelDir = startupPath + "\\inference\\ch_PP-OCRv4_rec_infer";
  59.                 labelFilePath = startupPath + "\\inference\\ppocr_keys.txt";
  60.             }
  61.             else
  62.             {
  63.                 detectionModelDir = startupPath + "\\inference_server\\detv4_teacher_inference";
  64.                 classificationModelDir = startupPath + "\\inference_server\\ch_ppocr_mobile_v2.0_cls_infer";
  65.                 recognitionModelDir = startupPath + "\\inference_server\\ch_PP-OCRv4_rec_server_infer";
  66.                 labelFilePath = startupPath + "\\inference_server\\ppocr_keys.txt";
  67.             }
  68.  
  69.             FullOcrModel model = FullOcrModel.FromDirectory(detectionModelDir, classificationModelDir, recognitionModelDir, labelFilePath, ModelVersion.V4);
  70.  
  71.             if (rbOnnx.Enabled == true)
  72.             {
  73.                 paddleOcr = new PaddleOcrAll(model, PaddleDevice.Onnx());
  74.             }
  75.  
  76.             else if (rbOpenblas.Enabled == true)
  77.             {
  78.                 paddleOcr = new PaddleOcrAll(model, PaddleDevice.Openblas());
  79.             }
  80.  
  81.             else if (rbMkldnn.Enabled == true)
  82.             {
  83.                 paddleOcr = new PaddleOcrAll(model, PaddleDevice.Mkldnn());
  84.             }
  85.             else
  86.             {
  87.                 paddleOcr = new PaddleOcrAll(model);
  88.             }
  89.  
  90.             paddleOcr.AllowRotateDetection = true;    /* 允许识别有角度的文字 */
  91.             paddleOcr.Enable180Classification = false; /* 允许识别旋转角度大于90度的文字 */
  92.         }
  93.  
  94.  
  95.         private void btnSelect_Click(object sender, EventArgs e)
  96.         {
  97.             if (paddleOcr == null)
  98.             {
  99.                 MessageBox.Show("请先加载模型");
  100.                 return;
  101.             }
  102.  
  103.  
  104.             OpenFileDialog ofd = new OpenFileDialog();
  105.             ofd.Filter = fileFilter;
  106.             if (ofd.ShowDialog() != DialogResult.OK) return;
  107.  
  108.             imgPath = ofd.FileName;
  109.             bmp = new Bitmap(imgPath, true);
  110.  
  111.             RotateImage(bmp);
  112.  
  113.             pictureBox1.Image = bmp;
  114.  
  115.             richTextBox1.Clear();
  116.  
  117.             mat = new Mat(ofd.FileName);
  118.             //pictureBox1.Image = OpenCvSharp.Extensions.BitmapConverter.ToBitmap(mat);
  119.  
  120.  
  121.             dt1 = DateTime.Now;
  122.             ocrResult = paddleOcr.Run(mat);
  123.             dt2 = DateTime.Now;
  124.  
  125.  
  126.             ShowOCRResult(ocrResult);
  127.  
  128.         }
  129.  
  130.         /// <summary>
  131.         /// 显示结果
  132.         /// </summary>
  133.         private void ShowOCRResult(PaddleOcrResult ocrResult)
  134.         {
  135.  
  136.             richTextBox1.Clear();
  137.             Bitmap bitmap = (Bitmap)this.pictureBox1.Image;
  138.  
  139.  
  140.             richTextBox1.AppendText("耗时:" + (dt2 - dt1).TotalMilliseconds + "ms\n");
  141.             richTextBox1.AppendText("-----------------------------------\n");
  142.             richTextBox1.AppendText(ocrResult.Text + "\n");
  143.  
  144.             using (Graphics g = Graphics.FromImage(bitmap))
  145.             {
  146.                 foreach (var item in ocrResult.Regions)
  147.                 {
  148.                     g.DrawRectangle(pen, item.Rect.BoundingRect().Left, item.Rect.BoundingRect().Top, item.Rect.BoundingRect().Width, item.Rect.BoundingRect().Height);
  149.                 }
  150.             }
  151.  
  152.             pictureBox1.Image = null;
  153.             pictureBox1.Image = bitmap;
  154.         }
  155.  
  156.         private void button1_Click(object sender, EventArgs e)
  157.         {
  158.             if (paddleOcr == null)
  159.             {
  160.                 MessageBox.Show("请先加载模型");
  161.                 return;
  162.             }
  163.  
  164.             if (pictureBox1.Image == null)
  165.             {
  166.                 return;
  167.             }
  168.             richTextBox1.Clear();
  169.             Application.DoEvents();
  170.  
  171.             dt1 = DateTime.Now;
  172.             ocrResult = paddleOcr.Run(mat);
  173.             dt2 = DateTime.Now;
  174.  
  175.             ShowOCRResult(ocrResult);
  176.         }
  177.  
  178.         /// <summary>
  179.         /// 根据图片exif调整方向
  180.         /// </summary>
  181.         /// <param name="img"></param>
  182.         public void RotateImage(Bitmap img)
  183.         {
  184.             var exif = img.PropertyItems;
  185.             byte orien = 0;
  186.             var item = exif.Where(m => m.Id == 274).ToArray();
  187.             if (item.Length > 0)
  188.                 orien = item[0].Value[0];
  189.             switch (orien)
  190.             {
  191.                 case 2:
  192.                     img.RotateFlip(RotateFlipType.RotateNoneFlipX);//horizontal flip
  193.                     break;
  194.                 case 3:
  195.                     img.RotateFlip(RotateFlipType.Rotate180FlipNone);//right-top
  196.                     break;
  197.                 case 4:
  198.                     img.RotateFlip(RotateFlipType.RotateNoneFlipY);//vertical flip
  199.                     break;
  200.                 case 5:
  201.                     img.RotateFlip(RotateFlipType.Rotate90FlipX);
  202.                     break;
  203.                 case 6:
  204.                     img.RotateFlip(RotateFlipType.Rotate90FlipNone);//right-top
  205.                     break;
  206.                 case 7:
  207.                     img.RotateFlip(RotateFlipType.Rotate270FlipX);
  208.                     break;
  209.                 case 8:
  210.                     img.RotateFlip(RotateFlipType.Rotate270FlipNone);//left-bottom
  211.                     break;
  212.                 default:
  213.                     break;
  214.             }
  215.         }
  216.  
  217.     }
  218. }

下载 

Demo下载

本文内容由网友自发贡献，转载请注明出处：https://www.wpsshop.cn/w/小小林熬夜学编程/article/detail/679049
推荐阅读
相关标签

Copyright © 2003-2013 www.wpsshop.cn 版权所有，并保留所有权利。

  

闽ICP备14008679号