C++ OCR证件照文字识别

 一.引言

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

二.简介

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

  • C++
  • Python
  • 易语言

1.C++头文件

  1. #ifndef __SN_OCR__H__
  2. #define __SN_OCR__H__
  3. #include "windows.h"
  4. //返回参数
  5. typedef struct SN_STATU {
  6. int code; //错误码,如果为 0 表示成功,否则表示错误号
  7. char message[4096]; //错误信息,如果为 "OK" 表示成功,否则返回错误信息
  8. }SN_STATU;
  9. /*启动OCR文字识别服务
  10. *
  11. * 参数:
  12. * [in] szOnnxFilePath: 设置 onnx 模型文件路径,如果设置为 NULL,默认和 DLL文件同级目录
  13. * [out] pResult: 返回错误信息,参数pResult->code(错误码)如果为 0 表示成功,否则表示错误号;
  14. *
  15. * 返回值:成功返回0,失败返回错误号,详细错误信息请参考 pResult
  16. *
  17. */
  18. int WINAPI apiSNInitOCRServer(char* szOnnxFilePath, SN_STATU* pStatu);
  19. /*创建OCR文字识别句柄
  20. *
  21. * 参数:
  22. * [in] szKey: 卡密(购买卡密:https://shop.4yuns.com/links/7C9F16B7)
  23. * [in] pOnnxFilePath:设置 onnx 模型文件路径,如果设置为 NULL,默认和 DLL文件同级目录
  24. * [out] pResult: 返回错误信息,参数pResult->code(错误码)如果为 0 表示成功,否则表示错误号;
  25. *
  26. * 返回值:成功返回句柄,失败返回NULL
  27. *
  28. */
  29. HANDLE WINAPI apiSNCreateOCRHandle(char* szKey, char* szOnnxFilePath, SN_STATU* pStatu);
  30. /*获取OCR文字识别卡密到期时间
  31. *
  32. * 参数:
  33. * [in] handle: 句柄(通过调用apiSNCreateOCRHandle得到)
  34. * [out] pResult: 返回错误信息,参数pResult->code(错误码)如果为 0 表示成功,否则表示错误号;
  35. *
  36. * 返回值:返回卡密到期时间,失败返回NULL,错误信息请查看参数 pResult->message
  37. *
  38. */
  39. char* WINAPI apiSNGetKeyExpiresTime(HANDLE handle, SN_STATU* pResult);
  40. /*获取OCR文字识别结果(以json字符串形式返回)
  41. *
  42. * 参数:
  43. * [in] handle: 句柄(通过调用apiSNCreateOCRHandle得到)
  44. * [in] szImageFilePath: 图片路径
  45. * [out] pResult: 返回错误信息,参数pResult->code(错误码)如果为 0 表示成功,否则表示错误号;
  46. *
  47. * 返回值:返回OCR文字识别结果(以json字符串形式返回),失败返回NULL,错误信息请查看参数 pResult->message
  48. *
  49. */
  50. char* WINAPI apiSNGetOCRFromImage(HANDLE handle, char* szImageFilePath, SN_STATU* pStatu);
  51. /*释放OCR文字识别句柄(释放内存)
  52. *
  53. * 参数:
  54. * [in] handle: 句柄(通过调用apiSNCreateOCRHandle得到)
  55. *
  56. * 返回值:返回 0 表示成功,其他值表示错误号;
  57. *
  58. */
  59. int WINAPI apiSNDestroyOCRHandle(HANDLE handle);
  60. #endif

2.C++调用dll接口

  1. #include <iostream>
  2. #include "SNOCR.h"
  3. int main()
  4. {
  5. struct SN_STATU statu = { 0 };
  6. char szExeFullPath[4096] = { 0 };
  7. char szImagePath[4096] = { 0 };
  8. //卡密
  9. char szKey[4096] = "SNKJe9xffLhdFY7r3TcffXq44ThDVcE3BQFQFfVA9VG4";
  10. //onnx模型路径
  11. char szOnnxFullPath[4096] = { 0 };
  12. GetModuleFileName(NULL, szExeFullPath, 4096);
  13. *strrchr(szExeFullPath, '\\') = 0;
  14. sprintf(szOnnxFullPath, "%s\\SNOCR.onnx", szExeFullPath);
  15. // 注意路径不要带有中文
  16. sprintf(szImagePath, "%s\\7.jpg", szExeFullPath);
  17. //1.启动OCR服务
  18. int ret = apiSNInitOCRServer(szOnnxFullPath, &statu);
  19. if (ret < 0)
  20. {
  21. printf("Error:%s \n", statu.message);
  22. return 0;
  23. }
  24. //2.创建OCR句柄
  25. HANDLE handle = apiSNCreateOCRHandle(szKey, szOnnxFullPath, &statu);
  26. if (!handle)
  27. {
  28. printf("Error:%s \n", statu.message);
  29. return 0;
  30. }
  31. //3.获取卡密到期时间
  32. char* szTime = apiSNGetKeyExpiresTime(handle, &statu);
  33. if (!szTime)
  34. {
  35. printf("Error:%s \n", statu.message);
  36. return 0;
  37. }
  38. //4.识别OCR,返回Json字符串
  39. char* szJson = apiSNGetOCRFromImage(handle, szImagePath, &statu);
  40. if (!szJson)
  41. {
  42. printf("Error:%s \n", statu.message);
  43. return 0;
  44. }
  45. printf("%s \n", szJson);
  46. //5.释放内存
  47. apiSNDestroyOCRHandle(handle);
  48. getchar();
  49. }

三.效果演示

1.图片1

识别效果:

  1. {
  2. "type": 0,
  3. "task_id": 1,
  4. "err_code": 0,
  5. "ocr_result": {
  6. "single_result": [{
  7. "left": 80.136589,
  8. "top": 56.710590,
  9. "right": 413.614105,
  10. "bottom": 89.287964,
  11. "str_utf8": "包头市特种设备追溯平台",
  12. "rate": "0.981197"
  13. }, {
  14. "left": 171.293091,
  15. "top": 99.701866,
  16. "right": 329.740753,
  17. "bottom": 120.792061,
  18. "str_utf8": "设备编码: 000001)",
  19. "rate": "0.970116"
  20. }, {
  21. "left": 81.693756,
  22. "top": 274.142029,
  23. "right": 229.766312,
  24. "bottom": 295.966248,
  25. "str_utf8": "RFID 扫描区域",
  26. "rate": "0.992770"
  27. }, {
  28. "left": 50,
  29. "top": 318.229156,
  30. "right": 181.250000,
  31. "bottom": 339.062500,
  32. "str_utf8": "投诉电话: 12365",
  33. "rate": "0.984698"
  34. }, {
  35. "left": 259.311310,
  36. "top": 352.951111,
  37. "right": 466.734924,
  38. "bottom": 371.130615,
  39. "str_utf8": "包头市质量技术监督局制",
  40. "rate": "0.961233"
  41. }],
  42. "width": "500",
  43. "height": "384"
  44. }
  45. }

2.图片2

识别效果:

  1. {
  2. "type": 0,
  3. "task_id": 1,
  4. "err_code": 0,
  5. "ocr_result": {
  6. "single_result": [{
  7. "left": 451.128448,
  8. "top": 110.489426,
  9. "right": 1138.148070,
  10. "bottom": 199.850967,
  11. "str_utf8": "中华人民共和国",
  12. "rate": "0.998395"
  13. }, {
  14. "left": 398.003052,
  15. "top": 250.290588,
  16. "right": 1189.906010,
  17. "bottom": 370.648926,
  18. "str_utf8": "居民身份证",
  19. "rate": "0.999714"
  20. }, {
  21. "left": 333.586945,
  22. "top": 605.802917,
  23. "right": 1028.648680,
  24. "bottom": 654.308594,
  25. "str_utf8": "签发机关上海市公安局徐汇分局",
  26. "rate": "0.998378"
  27. }, {
  28. "left": 334.754303,
  29. "top": 712.041199,
  30. "right": 539.191406,
  31. "bottom": 752.816345,
  32. "str_utf8": "有效期限",
  33. "rate": "0.999937"
  34. }, {
  35. "left": 551.186523,
  36. "top": 713.943665,
  37. "right": 1061.341670,
  38. "bottom": 754.974915,
  39. "str_utf8": "2005.10.08-202510.08",
  40. "rate": "0.985583"
  41. }],
  42. "width": "1313",
  43. "height": "858"
  44. }
  45. }

四.常见问题

1.是否支持多线程

支持

五.更新日志

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

六.云盘源码下载

posted @ 2024-12-17 07:08  猿说编程  阅读(37)  评论(0编辑  收藏  举报