1.所用到的ncnn格式的模型文件

 要问这些模型哪里来的,请看下面提示信息:

 2.查看字典函数读取方法

char* readKeysFromAssets()
{

	std::ifstream ifs("./model/paddleocr_keys.txt");
	if (!ifs.is_open())
	{
		return 0;
	}

	ifs.seekg(0, std::ios_base::end);
	int length = ifs.tellg();
	ifs.seekg(std::ios_base::beg);
	char* buff = new char[length + 1]();
	ifs.read(buff, length + 1);

	return buff;
}

这个paddleocr_keys.txt 的字符编码格式是可以修改,改为ANSI 格式,这样在cmd行中就不会中文乱码

 3.模型加载函数,分为检测模型和识别模型

int ret = dbNet.load_param("./model/pdocrv2.0_det-op.param");
		if (ret != 0)
		{
			return false;
		}
		ret = crnnNet.load_param("./model/pdocrv2.0_rec-op.param");
		if (ret != 0)
		{
			return false;
		}
	}
	// init bin
	{
		int ret = dbNet.load_model("./model/pdocrv2.0_det-op.bin");
		if (ret != 0)
		{
			return false;
		}
		ret = crnnNet.load_model("./model/pdocrv2.0_rec-op.bin");
		if (ret != 0)
		{
			return false;
		}
	}

4.模型推理主要函数

void PaddleOCRNcnn_Detect(bool use_gpu, cv::Mat& rgb)
{

	std::vector<TextBox> objects;
	objects = getTextBoxes(rgb, 0.4, 0.3, 2.0);

	std::vector<cv::Mat> partImages = getPartImages(rgb, objects);
	std::vector<TextLine> textLines = getTextLines(partImages);

	if (textLines.size() > 0)
	{
		for (int i = 0; i < textLines.size(); i++)
			objects[i].text = textLines[i].text;
	}

	if (objects.size() > 0)
	{
		for (size_t i = 0; i < objects.size(); i++)
		{

			float x0 = objects[i].boxPoint[0].x;
			float y0 = objects[i].boxPoint[0].y;
			float x1 = objects[i].boxPoint[1].x;
			float y1 = objects[i].boxPoint[1].y;
			float x2 = objects[i].boxPoint[2].x;
			float y2 = objects[i].boxPoint[2].y;
			float x3 = objects[i].boxPoint[3].x;
			float y3 = objects[i].boxPoint[3].y;

			cv::line(rgb, cv::Point(x0, y0), cv::Point(x1, y1), cv::Scalar(0, 0, 255), 2, 8);
			cv::line(rgb, cv::Point(x1, y1), cv::Point(x2, y2), cv::Scalar(0, 0, 255), 2, 8);
			cv::line(rgb, cv::Point(x2, y2), cv::Point(x3, y3), cv::Scalar(0, 0, 255), 2, 8);
			cv::line(rgb, cv::Point(x3, y3), cv::Point(x0, y0), cv::Scalar(0, 0, 255), 2, 8);

			cv::putTextZH(
				rgb,
				textLines[i].text.c_str(),
				cv::Point(x1, y1),
				CV_RGB(255, 255, 255),
				20
			);
		}
	}
}

5.OCR识别推理的结果

 我只是个代码搬运工,需要下载代码的地址。

https://download.csdn.net/download/huzhifei/88016666

GitHub 加速计划 / pa / PaddleOCR
41.53 K
7.59 K
下载
Awesome multilingual OCR toolkits based on PaddlePaddle (practical ultra lightweight OCR system, support 80+ languages recognition, provide data annotation and synthesis tools, support training and deployment among server, mobile, embedded and IoT devices)
最近提交(Master分支:3 个月前 )
7bbda2bc 6 天前
1d4e7a80 8 天前
Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐