C++ OCR证件照文字识别

发布于:2024-12-19 ⋅ 阅读:(10) ⋅ 点赞:(0)

 一.引言

文字识别,也称为光学字符识别(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

识别效果:

{
	"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

识别效果:

{
	"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/易语言

六.云盘源码下载