lw.PPOCRSharp_GPU_Test paddle_inference v3.3

目录

效果

项目

代码

下载


效果

NVIDIA GeForce RTX 4060 Laptop GPU 效果

项目

代码

using Newtonsoft.Json;

using OpenCvSharp;

using System;

using System.Collections.Generic;

using System.Diagnostics;

using System.Drawing;

using System.Runtime.InteropServices;

using System.Text;

using System.Windows.Forms;

namespace OCRV5Test

{

public partial class Form1 : Form

{

public Form1()

{

InitializeComponent();

}

const string DllName = "lw.PPOCRSharp.dll";

//初始化

DllImport(DllName, EntryPoint = "init", CallingConvention = CallingConvention.Cdecl)

public extern static int init(ref IntPtr engine

, bool use_gpu

, int gpu_id

, int gpu_mem

, int cpu_threads

, bool enable_mkldnn

, string det_model_dir

, int limit_side_len

, double det_db_thresh

, double det_db_box_thresh

, double det_db_unclip_ratio

, bool use_dilation

, bool cls

, bool use_angle_cls

, string cls_model_dir

, double cls_thresh

, double cls_batch_num

, string rec_model_dir

, string rec_char_dict_path

, int rec_batch_num

, int rec_img_h

, int rec_img_w

, int predictor_num

, StringBuilder msg);

//识别

DllImport(DllName, EntryPoint = "ocr", CallingConvention = CallingConvention.Cdecl)

public extern static int ocr(IntPtr engine, IntPtr image, StringBuilder msg, out IntPtr ocr_result, out int ocr_result_len);

//释放

DllImport(DllName, EntryPoint = "destroy", CallingConvention = CallingConvention.Cdecl)

public extern static int destroy(IntPtr engine, StringBuilder msg);

static IntPtr OCREngine;

private Bitmap bmp;

private String imgPath = null;

private List<OCRResult> ltOCRResult;

private string fileFilter = "*.*|*.bmp;*.jpg;*.jpeg;*.tiff;*.tif;*.png";

private StringBuilder OCRResultInfo = new StringBuilder();

private StringBuilder OCRResultAllInfo = new StringBuilder();

Pen pen = new Pen(Brushes.Red, 2f);

/// <summary>

/// 选择图片

/// </summary>

/// <param name="sender"></param>

/// <param name="e"></param>

private void button1_Click(object sender, EventArgs e)

{

OpenFileDialog ofd = new OpenFileDialog();

ofd.Filter = fileFilter;

if (ofd.ShowDialog() == DialogResult.OK)

{

imgPath = ofd.FileName;

bmp = new Bitmap(imgPath);

pictureBox1.Image = bmp;

richTextBox1.Clear();

button2_Click(null, null);

}

}

/// <summary>

/// 识别

/// </summary>

/// <param name="sender"></param>

/// <param name="e"></param>

private void button2_Click(object sender, EventArgs e)

{

if (OCREngine == IntPtr.Zero)

{

MessageBox.Show("请先初始化!!!");

return;

}

if (imgPath == null)

{

MessageBox.Show("请先选择图片!!!");

return;

}

button1.Enabled = false;

button2.Enabled = false;

richTextBox1.Clear();

OCRResultInfo.Clear();

OCRResultAllInfo.Clear();

Application.DoEvents();

Mat img = new Mat(imgPath);

StringBuilder msgTemp = new StringBuilder(128);

StringBuilder ocrResultStr = new StringBuilder(1024 * 100);

Stopwatch stopwatch = new Stopwatch();

stopwatch.Start();

IntPtr strPtr;

int ocr_result_len = 0;

int res = ocr(OCREngine, img.CvPtr, msgTemp, out strPtr, out ocr_result_len);

img.Dispose();

byte[] buffer = new byte[ocr_result_len];

Marshal.Copy(strPtr, buffer, 0, ocr_result_len);

string ocr_result = Encoding.UTF8.GetString(buffer);

Marshal.FreeCoTaskMem(strPtr);

stopwatch.Stop();

double totalTime = stopwatch.Elapsed.TotalMilliseconds;

OCRResultAllInfo.AppendLine($"耗时: {totalTime:F2}ms");

OCRResultAllInfo.AppendLine("---------------------------");

OCRResultInfo.AppendLine($"耗时: {totalTime:F2}ms");

OCRResultInfo.AppendLine("---------------------------");

if (res == 0)

{

ltOCRResult = Newtonsoft.Json.JsonConvert.DeserializeObject<List<OCRResult>>(ocr_result);

OCRResultAllInfo.Append(JsonConvert.SerializeObject(ltOCRResult, Newtonsoft.Json.Formatting.Indented));

Graphics graphics = Graphics.FromImage(bmp);

foreach (OCRResult item in ltOCRResult)

{

OCRResultInfo.AppendLine(item.text);

System.Drawing.Point[] pt = new System.Drawing.Point[] {

new System.Drawing.Point(item.x1, item.y1)

, new System.Drawing.Point(item.x2, item.y2)

, new System.Drawing.Point(item.x3, item.y3)

, new System.Drawing.Point(item.x4, item.y4)

};

graphics.DrawPolygon(pen, pt);

}

graphics.Dispose();

if (checkBox1.Checked)

{

richTextBox1.Text = OCRResultAllInfo.ToString();

}

else

{

richTextBox1.Text = OCRResultInfo.ToString();

}

pictureBox1.Image = null;

pictureBox1.Image = bmp;

}

else

{

MessageBox.Show("识别失败," + msgTemp.ToString());

}

img.Release();

button1.Enabled = true;

button2.Enabled = true;

}

/// <summary>

/// 初始化

/// </summary>

/// <param name="sender"></param>

/// <param name="e"></param>

private void Form1_Load(object sender, EventArgs e)

{

radioButton1.Checked = true;

chkUseGPU.Checked = true;

}

private void checkBox1_CheckedChanged(object sender, EventArgs e)

{

richTextBox1.Clear();

if (checkBox1.Checked)

{

richTextBox1.Text = OCRResultAllInfo.ToString();

}

else

{

richTextBox1.Text = OCRResultInfo.ToString();

}

}

private void radioButton1_CheckedChanged(object sender, EventArgs e)

{

RadioButton rb = sender as RadioButton;

}

private void Form1_FormClosing(object sender, FormClosingEventArgs e)

{

UnloadModel();

}

private void btnDestroy_Click(object sender, EventArgs e)

{

UnloadModel();

}

void UnloadModel()

{

if (OCREngine != IntPtr.Zero)

{

StringBuilder msgTemp = new StringBuilder(128);

destroy(OCREngine, msgTemp);

Console.WriteLine("释放成功:" + msgTemp.ToString());

OCREngine = IntPtr.Zero;

}

}

private void btnInit_Click(object sender, EventArgs e)

{

if (OCREngine != IntPtr.Zero)

{

StringBuilder msgTemp = new StringBuilder(128);

destroy(OCREngine, msgTemp);

Console.WriteLine("释放成功:" + msgTemp.ToString());

OCREngine = IntPtr.Zero;

LoadModel();

}

else {

LoadModel();

}

}

void LoadModel()

{

StringBuilder msgTemp = new StringBuilder(128);

string root_dir = Application.StartupPath + @"\inference";

bool use_gpu = true;

int gpu_id = 0;

int gpu_mem = 4000;

int cpu_threads = 10;

bool enable_mkldnn = true;

string det_model_dir = "";

int limit_side_len = 960;

double det_db_thresh = 0.3;

double det_db_box_thresh = 0.6;

double det_db_unclip_ratio = 1.2;

bool use_dilation = false;

bool cls = false;

bool use_angle_cls = true;

string cls_model_dir = "inference/ch_ppocr_mobile_v2.0_cls_infer/";

double cls_thresh = 0.9;

int cls_batch_num = 1;

string rec_model_dir = "";

string rec_char_dict_path = "inference/ppocrv5_dict.txt";

int rec_batch_num = 8;

int rec_img_h = 48;

int rec_img_w = 320;

int predictor_num = 4;

//赋值

if (chkUseGPU.Checked == true)

{

use_gpu = true;

}

else

{

use_gpu = false;

}

if (chkEnableMkldnn.Checked == true)

{

enable_mkldnn = true;

}

else

{

enable_mkldnn = false;

}

cpu_threads = Convert.ToInt32(txtcpu_threads.Text.ToString());

rec_batch_num = Convert.ToInt32(txtrec_batch_num.Text.ToString());

rec_img_h = Convert.ToInt32(txtrec_img_h.Text.ToString());

rec_img_w = Convert.ToInt32(txtrec_img_w.Text.ToString());

predictor_num = Convert.ToInt32(txtpredictor_num.Text.ToString());

if (radioButton1.Checked)

{

det_model_dir = "inference/PP-OCRv5_mobile_det_infer/";

rec_model_dir = "inference/PP-OCRv5_mobile_rec_infer/";

}

else

{

det_model_dir = "inference/PP-OCRv5_server_det_infer/";

rec_model_dir = "inference/PP-OCRv5_server_rec_infer/";

}

int res = init(ref OCREngine

, use_gpu

, gpu_id

, gpu_mem

, cpu_threads

, enable_mkldnn

, det_model_dir

, limit_side_len

, det_db_thresh

, det_db_box_thresh

, det_db_unclip_ratio

, use_dilation

, cls

, use_angle_cls

, cls_model_dir

, cls_thresh

, cls_batch_num

, rec_model_dir

, rec_char_dict_path

, rec_batch_num

, rec_img_h

, rec_img_w

, predictor_num

, msgTemp);

if (res == 0)

{

Console.WriteLine("模型加载成功!");

}

else

{

string msg = msgTemp.ToString();

Console.WriteLine("模型加载失败," + msg);

}

}

}

}

复制代码
using Newtonsoft.Json;
using OpenCvSharp;
using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.Drawing;
using System.Runtime.InteropServices;
using System.Text;
using System.Windows.Forms;

namespace OCRV5Test
{
    public partial class Form1 : Form
    {
        public Form1()
        {
            InitializeComponent();
        }

        const string DllName = "lw.PPOCRSharp.dll";

        //初始化
        [DllImport(DllName, EntryPoint = "init", CallingConvention = CallingConvention.Cdecl)]
        public extern static int init(ref IntPtr engine

            , bool use_gpu
            , int gpu_id
            , int gpu_mem

            , int cpu_threads
            , bool enable_mkldnn

            , string det_model_dir
            , int limit_side_len
            , double det_db_thresh
            , double det_db_box_thresh
            , double det_db_unclip_ratio
            , bool use_dilation

            , bool cls
            , bool use_angle_cls
            , string cls_model_dir
            , double cls_thresh
            , double cls_batch_num

            , string rec_model_dir
            , string rec_char_dict_path
            , int rec_batch_num
            , int rec_img_h
            , int rec_img_w
            , int predictor_num

            , StringBuilder msg);

        //识别
        [DllImport(DllName, EntryPoint = "ocr", CallingConvention = CallingConvention.Cdecl)]
        public extern static int ocr(IntPtr engine, IntPtr image, StringBuilder msg, out IntPtr ocr_result, out int ocr_result_len);

        //释放
        [DllImport(DllName, EntryPoint = "destroy", CallingConvention = CallingConvention.Cdecl)]
        public extern static int destroy(IntPtr engine, StringBuilder msg);

        static IntPtr OCREngine;

        private Bitmap bmp;

        private String imgPath = null;

        private List<OCRResult> ltOCRResult;

        private string fileFilter = "*.*|*.bmp;*.jpg;*.jpeg;*.tiff;*.tif;*.png";

        private StringBuilder OCRResultInfo = new StringBuilder();
        private StringBuilder OCRResultAllInfo = new StringBuilder();

        Pen pen = new Pen(Brushes.Red, 2f);

        /// <summary>
        /// 选择图片
        /// </summary>
        /// <param name="sender"></param>
        /// <param name="e"></param>
        private void button1_Click(object sender, EventArgs e)
        {
            OpenFileDialog ofd = new OpenFileDialog();
            ofd.Filter = fileFilter;
            if (ofd.ShowDialog() == DialogResult.OK)
            {
                imgPath = ofd.FileName;
                bmp = new Bitmap(imgPath);
                pictureBox1.Image = bmp;
                richTextBox1.Clear();

                button2_Click(null, null);
            }
        }

        /// <summary>
        /// 识别
        /// </summary>
        /// <param name="sender"></param>
        /// <param name="e"></param>
        private void button2_Click(object sender, EventArgs e)
        {
            if (OCREngine == IntPtr.Zero)
            {
                MessageBox.Show("请先初始化!!!");
                return;
            }

            if (imgPath == null)
            {
                MessageBox.Show("请先选择图片!!!");
                return;
            }

            button1.Enabled = false;
            button2.Enabled = false;
            richTextBox1.Clear();
            OCRResultInfo.Clear();
            OCRResultAllInfo.Clear();

            Application.DoEvents();
            Mat img = new Mat(imgPath);
            StringBuilder msgTemp = new StringBuilder(128);
            StringBuilder ocrResultStr = new StringBuilder(1024 * 100);

            Stopwatch stopwatch = new Stopwatch();
            stopwatch.Start();

            IntPtr strPtr;
            int ocr_result_len = 0;

            int res = ocr(OCREngine, img.CvPtr, msgTemp, out strPtr, out ocr_result_len);
            img.Dispose();
            byte[] buffer = new byte[ocr_result_len];
            Marshal.Copy(strPtr, buffer, 0, ocr_result_len);
            string ocr_result = Encoding.UTF8.GetString(buffer);
            Marshal.FreeCoTaskMem(strPtr);
            stopwatch.Stop();
            double totalTime = stopwatch.Elapsed.TotalMilliseconds;

            OCRResultAllInfo.AppendLine($"耗时: {totalTime:F2}ms");
            OCRResultAllInfo.AppendLine("---------------------------");

            OCRResultInfo.AppendLine($"耗时: {totalTime:F2}ms");
            OCRResultInfo.AppendLine("---------------------------");

            if (res == 0)
            {
                ltOCRResult = Newtonsoft.Json.JsonConvert.DeserializeObject<List<OCRResult>>(ocr_result);
                OCRResultAllInfo.Append(JsonConvert.SerializeObject(ltOCRResult, Newtonsoft.Json.Formatting.Indented));
                Graphics graphics = Graphics.FromImage(bmp);

                foreach (OCRResult item in ltOCRResult)
                {
                    OCRResultInfo.AppendLine(item.text);
                    System.Drawing.Point[] pt = new System.Drawing.Point[] {
                              new System.Drawing.Point(item.x1, item.y1)
                            , new System.Drawing.Point(item.x2, item.y2)
                            , new System.Drawing.Point(item.x3, item.y3)
                            , new System.Drawing.Point(item.x4, item.y4)
                        };
                    graphics.DrawPolygon(pen, pt);
                }
                graphics.Dispose();

                if (checkBox1.Checked)
                {
                    richTextBox1.Text = OCRResultAllInfo.ToString();
                }
                else
                {
                    richTextBox1.Text = OCRResultInfo.ToString();
                }

                pictureBox1.Image = null;
                pictureBox1.Image = bmp;
            }
            else
            {
                MessageBox.Show("识别失败," + msgTemp.ToString());
            }

            img.Release();
            button1.Enabled = true;
            button2.Enabled = true;
        }

        /// <summary>
        /// 初始化
        /// </summary>
        /// <param name="sender"></param>
        /// <param name="e"></param>
        private void Form1_Load(object sender, EventArgs e)
        {
            radioButton1.Checked = true;
            chkUseGPU.Checked = true;
        }

        private void checkBox1_CheckedChanged(object sender, EventArgs e)
        {
            richTextBox1.Clear();
            if (checkBox1.Checked)
            {
                richTextBox1.Text = OCRResultAllInfo.ToString();
            }
            else
            {
                richTextBox1.Text = OCRResultInfo.ToString();
            }
        }

        private void radioButton1_CheckedChanged(object sender, EventArgs e)
        {
            RadioButton rb = sender as RadioButton;
        }

        private void Form1_FormClosing(object sender, FormClosingEventArgs e)
        {
            UnloadModel();
        }

        private void btnDestroy_Click(object sender, EventArgs e)
        {
            UnloadModel();
        }
     

        void UnloadModel()
        {
            if (OCREngine != IntPtr.Zero)
            {
                StringBuilder msgTemp = new StringBuilder(128);
                destroy(OCREngine, msgTemp);

                Console.WriteLine("释放成功:" + msgTemp.ToString());

                OCREngine = IntPtr.Zero;
            }
        }

        private void btnInit_Click(object sender, EventArgs e)
        {
            if (OCREngine != IntPtr.Zero)
            {
                StringBuilder msgTemp = new StringBuilder(128);
                destroy(OCREngine, msgTemp);

                Console.WriteLine("释放成功:" + msgTemp.ToString());

                OCREngine = IntPtr.Zero;

                LoadModel();
            }
            else {

                LoadModel();
            }
        }

        void LoadModel()
        {
            StringBuilder msgTemp = new StringBuilder(128);
            string root_dir = Application.StartupPath + @"\inference";

            bool use_gpu = true;

            int gpu_id = 0;
            int gpu_mem = 4000;

            int cpu_threads = 10;
            bool enable_mkldnn = true;

            string det_model_dir = "";
            int limit_side_len = 960;
            double det_db_thresh = 0.3;
            double det_db_box_thresh = 0.6;
            double det_db_unclip_ratio = 1.2;
            bool use_dilation = false;

            bool cls = false;
            bool use_angle_cls = true;
            string cls_model_dir = "inference/ch_ppocr_mobile_v2.0_cls_infer/";
            double cls_thresh = 0.9;
            int cls_batch_num = 1;

            string rec_model_dir = "";
            string rec_char_dict_path = "inference/ppocrv5_dict.txt";
            int rec_batch_num = 8;
            int rec_img_h = 48;
            int rec_img_w = 320;
            int predictor_num = 4;


            //赋值
            if (chkUseGPU.Checked == true)
            {
                use_gpu = true;
            }
            else
            {
                use_gpu = false;
            }

            if (chkEnableMkldnn.Checked == true)
            {
                enable_mkldnn = true;
            }
            else
            {
                enable_mkldnn = false;
            }

            cpu_threads = Convert.ToInt32(txtcpu_threads.Text.ToString());
            rec_batch_num = Convert.ToInt32(txtrec_batch_num.Text.ToString());
            rec_img_h = Convert.ToInt32(txtrec_img_h.Text.ToString());
            rec_img_w = Convert.ToInt32(txtrec_img_w.Text.ToString());
            predictor_num = Convert.ToInt32(txtpredictor_num.Text.ToString());

            if (radioButton1.Checked)
            {
                det_model_dir = "inference/PP-OCRv5_mobile_det_infer/";
                rec_model_dir = "inference/PP-OCRv5_mobile_rec_infer/";
            }
            else
            {
                det_model_dir = "inference/PP-OCRv5_server_det_infer/";
                rec_model_dir = "inference/PP-OCRv5_server_rec_infer/";
            }
            int res = init(ref OCREngine

                        , use_gpu
                        , gpu_id
                        , gpu_mem

                        , cpu_threads
                        , enable_mkldnn

                        , det_model_dir
                        , limit_side_len
                        , det_db_thresh
                        , det_db_box_thresh
                        , det_db_unclip_ratio
                        , use_dilation

                        , cls
                        , use_angle_cls
                        , cls_model_dir
                        , cls_thresh
                        , cls_batch_num

                        , rec_model_dir
                        , rec_char_dict_path
                        , rec_batch_num
                        , rec_img_h
                        , rec_img_w
                        , predictor_num

                        , msgTemp);

            if (res == 0)
            {
                Console.WriteLine("模型加载成功!");
            }
            else
            {
                string msg = msgTemp.ToString();
                Console.WriteLine("模型加载失败," + msg);
            }
        }
    }

}

下载

通过网盘分享的文件:lw.PPOCRSharp_GPU_Test (天天代码码天天) paddle_inference v3.3

链接: https://pan.baidu.com/s/1pdv7SEjJPwhfZQrVOCRo8w?pwd=5bf8 提取码: 5bf8

相关推荐
钮钴禄·爱因斯晨2 小时前
机器学习(三):聚焦KNN算法距离度量、特征预处理与超参数选择
人工智能·算法·机器学习
茶栀(*´I`*)2 小时前
PyTorch实战:用RNN从零开始实现周杰伦歌词生成
pytorch·rnn·深度学习
HZjiangzi2 小时前
盾构机刀盘磨损三维测量技术与思看科技SIMSCAN解决方案
人工智能·科技·3d
赵部长风向标2 小时前
【无标题】
人工智能
龙智DevSecOps解决方案2 小时前
现代服务管理指南:Jira Service Management + Rovo的AI自动化架构与实战应用
人工智能·自动化·atlassian·jira·itsm·服务管理
爱喝可乐的老王2 小时前
神经网络的学习
人工智能·神经网络·学习
阿里巴巴与四十个小矮人2 小时前
国科大2025秋自然语言处理基础与大模型期末
人工智能·自然语言处理
yumgpkpm2 小时前
在AI语言大模型时代 Cloudera CDP(华为CMP 鲲鹏版)对自有知识的保护
人工智能·hadoop·华为·zookeeper·spark·kafka
沃达德软件2 小时前
巡防勤务可视化管理
大数据·人工智能·数据挖掘·数据分析