C++ OCR证件照文字识别

一.引言

文字识别,也称为光学字符识别(Optical Character Recognition, OCR),是一种将不同形式的文档(如扫描的纸质文档、PDF文件或数字相机拍摄的图片)中的文字转换成可编辑和可搜索的数据的技术。随着技术的发展,文字识别技术已经成为信息管理、自动化办公和智能系统的关键组成部分。

二.简介

为了易于集成和使用,我们将文字识别OCR封装为DLL(动态链接库)。这种封装方式不仅保留了算法的性能优势,还提供了跨平台和跨语言的兼容性,目前支持编程语言如下:

  • C++
  • Python
  • 易语言

1.C++头文件


#ifndef __SN_OCR__H__
#define __SN_OCR__H__

#include "windows.h"

//返回参数
typedef struct SN_STATU {

    int code;           //错误码,如果为 0 表示成功,否则表示错误号
    char message[4096]; //错误信息,如果为 "OK" 表示成功,否则返回错误信息

}SN_STATU;

/*启动OCR文字识别服务
*
* 参数:
*   [in]  szOnnxFilePath:   设置 onnx 模型文件路径,如果设置为 NULL,默认和 DLL文件同级目录
*   [out] pResult:          返回错误信息,参数pResult->code(错误码)如果为 0 表示成功,否则表示错误号;
*
* 返回值:成功返回0,失败返回错误号,详细错误信息请参考 pResult
*
*/
int WINAPI apiSNInitOCRServer(char* szOnnxFilePath, SN_STATU* pStatu);

/*创建OCR文字识别句柄
*
* 参数:
*   [in]  szKey:        卡密(购买卡密:https://shop.4yuns.com/links/7C9F16B7)
*   [in]  pOnnxFilePath:设置 onnx 模型文件路径,如果设置为 NULL,默认和 DLL文件同级目录
*   [out] pResult:      返回错误信息,参数pResult->code(错误码)如果为 0 表示成功,否则表示错误号;
*
* 返回值:成功返回句柄,失败返回NULL
*
*/
HANDLE WINAPI apiSNCreateOCRHandle(char* szKey, char* szOnnxFilePath, SN_STATU* pStatu);

/*获取OCR文字识别卡密到期时间
*
* 参数:
*   [in]  handle:       句柄(通过调用apiSNCreateOCRHandle得到)
*   [out] pResult:      返回错误信息,参数pResult->code(错误码)如果为 0 表示成功,否则表示错误号;
*
* 返回值:返回卡密到期时间,失败返回NULL,错误信息请查看参数 pResult->message
*
*/
char* WINAPI apiSNGetKeyExpiresTime(HANDLE handle, SN_STATU* pResult);

/*获取OCR文字识别结果(以json字符串形式返回)
*
* 参数:
*   [in]  handle:           句柄(通过调用apiSNCreateOCRHandle得到)
*   [in]  szImageFilePath:  图片路径
*   [out] pResult:          返回错误信息,参数pResult->code(错误码)如果为 0 表示成功,否则表示错误号;
*
* 返回值:返回OCR文字识别结果(以json字符串形式返回),失败返回NULL,错误信息请查看参数 pResult->message
*
*/
char* WINAPI apiSNGetOCRFromImage(HANDLE handle, char* szImageFilePath, SN_STATU* pStatu);

/*释放OCR文字识别句柄(释放内存)
*
* 参数:
*   [in] handle:        句柄(通过调用apiSNCreateOCRHandle得到)
*
* 返回值:返回 0 表示成功,其他值表示错误号;
*
*/
int WINAPI apiSNDestroyOCRHandle(HANDLE handle);

#endif

2.C++调用dll接口


#include <iostream>
#include "SNOCR.h"

int main()
{
    struct SN_STATU statu = { 0 };

    char szExeFullPath[4096] = { 0 };
    char szImagePath[4096] = { 0 };
    //卡密
    char szKey[4096] = "SNKJe9xffLhdFY7r3TcffXq44ThDVcE3BQFQFfVA9VG4";
    //onnx模型路径
    char szOnnxFullPath[4096] = { 0 };

    GetModuleFileName(NULL, szExeFullPath, 4096);
    *strrchr(szExeFullPath, '\\') = 0;
    sprintf(szOnnxFullPath, "%s\\SNOCR.onnx", szExeFullPath);
    // 注意路径不要带有中文
    sprintf(szImagePath, "%s\\7.jpg", szExeFullPath);

    //1.启动OCR服务
    int ret = apiSNInitOCRServer(szOnnxFullPath, &statu);
    if (ret < 0)
    {
        printf("Error:%s \n", statu.message);
        return 0;
    }

    //2.创建OCR句柄
    HANDLE handle = apiSNCreateOCRHandle(szKey, szOnnxFullPath, &statu);
    if (!handle)
    {
        printf("Error:%s \n", statu.message);
        return 0;
    }

    //3.获取卡密到期时间
    char* szTime = apiSNGetKeyExpiresTime(handle, &statu);
    if (!szTime)
    {
        printf("Error:%s \n", statu.message);
        return 0;
    }

    //4.识别OCR,返回Json字符串
    char* szJson = apiSNGetOCRFromImage(handle, szImagePath, &statu);
    if (!szJson)
    {
        printf("Error:%s \n", statu.message);
        return 0;
    }
    printf("%s \n", szJson);
    //5.释放内存
    apiSNDestroyOCRHandle(handle);

    getchar();
}

三.效果演示

1.图片1

22.jpg

识别效果:

{
    "type": 0,
    "task_id":  1,
    "err_code": 0,
    "ocr_result":   {
        "single_result":    [{
                "left": 80.136589,
                "top":  56.710590,
                "right":    413.614105,
                "bottom":   89.287964,
                "str_utf8": "包头市特种设备追溯平台",
                "rate": "0.981197"
            }, {
                "left": 171.293091,
                "top":  99.701866,
                "right":    329.740753,
                "bottom":   120.792061,
                "str_utf8": "设备编码: 000001)",
                "rate": "0.970116"
            }, {
                "left": 81.693756,
                "top":  274.142029,
                "right":    229.766312,
                "bottom":   295.966248,
                "str_utf8": "RFID 扫描区域",
                "rate": "0.992770"
            }, {
                "left": 50,
                "top":  318.229156,
                "right":    181.250000,
                "bottom":   339.062500,
                "str_utf8": "投诉电话: 12365",
                "rate": "0.984698"
            }, {
                "left": 259.311310,
                "top":  352.951111,
                "right":    466.734924,
                "bottom":   371.130615,
                "str_utf8": "包头市质量技术监督局制",
                "rate": "0.961233"
            }],
        "width":    "500",
        "height":   "384"
    }
}

2.图片2

06.jpg

识别效果:

{
    "type": 0,
    "task_id":  1,
    "err_code": 0,
    "ocr_result":   {
        "single_result":    [{
                "left": 451.128448,
                "top":  110.489426,
                "right":    1138.148070,
                "bottom":   199.850967,
                "str_utf8": "中华人民共和国",
                "rate": "0.998395"
            }, {
                "left": 398.003052,
                "top":  250.290588,
                "right":    1189.906010,
                "bottom":   370.648926,
                "str_utf8": "居民身份证",
                "rate": "0.999714"
            }, {
                "left": 333.586945,
                "top":  605.802917,
                "right":    1028.648680,
                "bottom":   654.308594,
                "str_utf8": "签发机关上海市公安局徐汇分局",
                "rate": "0.998378"
            }, {
                "left": 334.754303,
                "top":  712.041199,
                "right":    539.191406,
                "bottom":   752.816345,
                "str_utf8": "有效期限",
                "rate": "0.999937"
            }, {
                "left": 551.186523,
                "top":  713.943665,
                "right":    1061.341670,
                "bottom":   754.974915,
                "str_utf8": "2005.10.08-202510.08",
                "rate": "0.985583"
            }],
        "width":    "1313",
        "height":   "858"
    }
}

四.常见问题

1.是否支持多线程

支持

五.更新日志

  • 2024.12.15 OCR 文字识别支持C++/Python/易语言

六.云盘源码下载

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容