My-PaddleOCR

介绍

如何在 C++ 项目中,通过源码使用 PaddlePaddle 实现 OCR 功能。

本项目的所有源码:gitee: paddleocr

目前,官方提供使用 PaddleOcr 的方案有:

  • 在 Python 项目中,调用 paddlepaddle + paddleocr 包。

  • 在 C++项目中,调用一个可执行文件。(由编译 PaddleOCR 中的 deploy/cpp_infer 下的代码形成)

    Paddle OCR 提供了一个通过编译 deploy/cpp_infer 下的代码为 ppocr.exe,然后通过命令行调用获取 OCR 的结果。

    具体过程见: 服务器端 C++预测

其它方法:

  • 使用 Python 写一个 RESTful 服务,然后让 C++项目调用这个服务功能。

这里主要介绍一个更加直接的方法:

  • deploy/cpp_infer 的源码引入到我们的 C++项目。
  • 做适当修改后,我们就可以直接使用这些 API。
  • 当然,如果你愿意的话,也可以将这些源码形成一个新的项目,编译成一个 dll。(这里并不介绍)

    说明:
  • 下面的方法只在 release 版中有效。可能是因为paddle_inference.dll 只支持 release 版。

    如果要支持 debug 版,可能需要重新编译 PaddlePaddle。

Paddle OCR C++ 源码

Paddle OCR 的仓库,在github: PaddleOCR 或者 gitee: PaddleOCR

C++ 相关的代码在目录 deploy/cpp_infer 里。

如何引入 Paddle OCR C++ 源码

需要安装的组件

  • opencv

    我在 opencv 4.6 版本上测试通过。

    注意:opencv 4.5 版本存在一些问题,会导致功能异常。

    设置环境变量:
@REM 设置opencv目录环境变量
setx OPENCV_ROOT "C:\3rd\opencv4.6\build"
  • paddle_inference

    我用的是 2.6 版。

    设置环境变量:
@REM 设置paddle_inference目录环境变量
setx PADDLE_ROOT "C:\3rd\paddle_inference"
  • cuda (optional)

    你应该懂得。
  • cudnn (optional)

    你应该懂得。

创建 C++项目

  • 创建一个空的 C++ 项目,并在项目中添加 opencv、paddle_inference、cuda、cudnn。

  • 在项目中创建文件:OpenCVDebug.props

<?xml version="1.0" encoding="utf-8"?>
<Project ToolsVersion="4.0" xmlns="http://schemas.microsoft.com/developer/msbuild/2003">
<ImportGroup Label="PropertySheets" />
<PropertyGroup Label="UserMacros" />
<PropertyGroup />
<ItemDefinitionGroup>
<ClCompile>
<AdditionalIncludeDirectories>$(OPENCV_ROOT)\include;%(AdditionalIncludeDirectories)</AdditionalIncludeDirectories>
</ClCompile>
<Link>
<AdditionalLibraryDirectories>$(OPENCV_ROOT)\x64\vc15\lib;%(AdditionalLibraryDirectories)</AdditionalLibraryDirectories>
<AdditionalDependencies>opencv_world460d.lib;%(AdditionalDependencies)</AdditionalDependencies>
</Link>
</ItemDefinitionGroup>
<ItemGroup />
</Project>
  • 在项目中创建文件:OpenCVRelease.props
<?xml version="1.0" encoding="utf-8"?>
<Project ToolsVersion="4.0" xmlns="http://schemas.microsoft.com/developer/msbuild/2003">
<ImportGroup Label="PropertySheets" />
<PropertyGroup Label="UserMacros" />
<PropertyGroup />
<ItemDefinitionGroup>
<ClCompile>
<AdditionalIncludeDirectories>$(OPENCV_ROOT)\include;%(AdditionalIncludeDirectories)</AdditionalIncludeDirectories>
</ClCompile>
<Link>
<AdditionalLibraryDirectories>$(OPENCV_ROOT)\x64\vc15\lib;%(AdditionalLibraryDirectories)</AdditionalLibraryDirectories>
<AdditionalDependencies>opencv_world460.lib;%(AdditionalDependencies)</AdditionalDependencies>
</Link>
</ItemDefinitionGroup>
<ItemGroup />
</Project>
  • 在项目中创建文件:PaddleDebug.props
<?xml version="1.0" encoding="utf-8"?>
<Project ToolsVersion="4.0" xmlns="http://schemas.microsoft.com/developer/msbuild/2003">
<ImportGroup Label="PropertySheets" />
<PropertyGroup Label="UserMacros" />
<PropertyGroup />
<ItemDefinitionGroup>
<ClCompile>
<AdditionalIncludeDirectories>./;$(PADDLE_ROOT)/paddle/include;$(PADDLE_ROOT)/third_party/install/zlib/include;$(PADDLE_ROOT)/third_party/boost;$(PADDLE_ROOT)/third_party/eigen3;$(PADDLE_ROOT)/third_party/install/mklml/include;%(AdditionalIncludeDirectories)</AdditionalIncludeDirectories>
</ClCompile>
<Link>
<AdditionalLibraryDirectories>$(PADDLE_ROOT)/third_party/install/zlib/lib;$(PADDLE_ROOT)/third_party/install/mklml/lib;$(PADDLE_ROOT)/paddle/lib;%(AdditionalLibraryDirectories)</AdditionalLibraryDirectories>
<AdditionalDependencies>paddle_inference.lib;mklml.lib;libiomp5md.lib;%(AdditionalDependencies)</AdditionalDependencies>
</Link>
</ItemDefinitionGroup>
<ItemGroup />
</Project>
  • 在项目中创建文件:PaddleRelease.props
<?xml version="1.0" encoding="utf-8"?>
<Project ToolsVersion="4.0" xmlns="http://schemas.microsoft.com/developer/msbuild/2003">
<ImportGroup Label="PropertySheets" />
<PropertyGroup Label="UserMacros" />
<PropertyGroup />
<ItemDefinitionGroup>
<ClCompile>
<AdditionalIncludeDirectories>./;$(PADDLE_ROOT)/paddle/include;$(PADDLE_ROOT)/third_party/install/zlib/include;$(PADDLE_ROOT)/third_party/boost;$(PADDLE_ROOT)/third_party/eigen3;$(PADDLE_ROOT)/third_party/install/mklml/include;%(AdditionalIncludeDirectories)</AdditionalIncludeDirectories>
</ClCompile>
<Link>
<AdditionalLibraryDirectories>$(PADDLE_ROOT)/third_party/install/zlib/lib;$(PADDLE_ROOT)/third_party/install/mklml/lib;$(PADDLE_ROOT)/paddle/lib;%(AdditionalLibraryDirectories)</AdditionalLibraryDirectories>
<AdditionalDependencies>paddle_inference.lib;mklml.lib;libiomp5md.lib;%(AdditionalDependencies)</AdditionalDependencies>
</Link>
</ItemDefinitionGroup>
<ItemGroup />
</Project>
  • 修改 C++项目文件:my-paddleocr.vcxproj
  <ImportGroup Label="PropertySheets" Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">
<Import Project="$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props" Condition="exists('$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props')" Label="LocalAppDataPlatform" />
<Import Project=".\OpenCVDebug.props" />
<Import Project=".\PaddleDebug.props" />
</ImportGroup>
<ImportGroup Label="PropertySheets" Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">
<Import Project="$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props" Condition="exists('$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props')" Label="LocalAppDataPlatform" />
<Import Project=".\OpenCVRelease.props" />
<Import Project=".\PaddleRelease.props" />
</ImportGroup>
<ImportGroup Label="PropertySheets" Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">
<Import Project="$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props" Condition="exists('$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props')" Label="LocalAppDataPlatform" />
<Import Project=".\OpenCVDebug.props" />
<Import Project=".\PaddleDebug.props" />
</ImportGroup>
<ImportGroup Label="PropertySheets" Condition="'$(Configuration)|$(Platform)'=='Release|x64'">
<Import Project="$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props" Condition="exists('$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props')" Label="LocalAppDataPlatform" />
<Import Project=".\OpenCVRelease.props" />
<Import Project=".\PaddleRelease.props" />
</ImportGroup>

引入 Paddle OCR C++ 源码

  • 克隆 Paddle OCR 仓库到本地。
  • 切换到分支 release/2.6
git clone https://gitee.com/paddlepaddle/PaddleOCR.git
cd PaddleOCR
git checkout release/2.6
  • deploy/cpp_infer 目录下的 includesrc两个目录的内容复制到我们的 C++ 项目中。
  • 修改新的 src 目录名称为 ocr
  • 删除ocr/main.cpp
  • include目录下创建ocr_flags.h文件,内容如下:

    这文件是为了替换 google 的 gflags 库的使用。
#pragma once

// common args
#include <string>
using std::string; #define DECLARE_bool(name) extern bool FLAGS_##name;
#define DECLARE_int32(name) extern int FLAGS_##name;
#define DECLARE_string(name) extern string FLAGS_##name;
#define DECLARE_double(name) extern double FLAGS_##name; #define DEFINE_VARIABLE(type, name, value, help) \
static const type FLAGS_nono##name = value; \
type FLAGS_##name = FLAGS_nono##name; \
static type FLAGS_no##name = FLAGS_nono##name; #define DEFINE_bool(name, val, txt) DEFINE_VARIABLE(bool, name, val, txt)
#define DEFINE_int32(name, val, txt) DEFINE_VARIABLE(int, name, val, txt)
#define DEFINE_string(name, val, txt) DEFINE_VARIABLE(string, name, val, txt)
#define DEFINE_double(name, val, txt) DEFINE_VARIABLE(double, name, val, txt)
  • 修改include/args.h文件,内容如下:

    不使用 gflags/gflags.h

    注释行 // #include <gflags/gflags.h>

    增加行 #include "ocr_flags.h"
...
// #include <gflags/gflags.h>
#include "ocr_flags.h"
...
  • 修改ocr/args.cpp文件,内容如下:

    不使用 gflags/gflags.h

    注释行 // #include <gflags/gflags.h>

    增加行 #include "include/ocr_flags.h"
...
// #include <gflags/gflags.h>
#include "include/ocr_flags.h"
...
  • 修改include/paddleocr.cpp文件,内容如下:

    不使用 auto_log/autolog.h

    注释行 // #include "auto_log/autolog.h"

    注释方法 PPOCR::benchmark_log 的内容。
// #include "auto_log/autolog.h"
...
void PPOCR::benchmark_log(int img_num) {
// if (this->time_info_det[0] + this->time_info_det[1] +
// this->time_info_det[2] >
// 0) {
// AutoLogger autolog_det("ocr_det", FLAGS_use_gpu, FLAGS_use_tensorrt,
// FLAGS_enable_mkldnn, FLAGS_cpu_threads, 1,
// "dynamic", FLAGS_precision, this->time_info_det,
// img_num);
// autolog_det.report();
// }
// if (this->time_info_rec[0] + this->time_info_rec[1] +
// this->time_info_rec[2] >
// 0) {
// AutoLogger autolog_rec("ocr_rec", FLAGS_use_gpu, FLAGS_use_tensorrt,
// FLAGS_enable_mkldnn, FLAGS_cpu_threads,
// FLAGS_rec_batch_num, "dynamic", FLAGS_precision,
// this->time_info_rec, img_num);
// autolog_rec.report();
// }
// if (this->time_info_cls[0] + this->time_info_cls[1] +
// this->time_info_cls[2] >
// 0) {
// AutoLogger autolog_cls("ocr_cls", FLAGS_use_gpu, FLAGS_use_tensorrt,
// FLAGS_enable_mkldnn, FLAGS_cpu_threads,
// FLAGS_cls_batch_num, "dynamic", FLAGS_precision,
// this->time_info_cls, img_num);
// autolog_cls.report();
// }
}
...
  • 修改include/paddlestructure.cpp文件,内容如下:

    不使用 auto_log/autolog.h

    注释行 // #include "auto_log/autolog.h"

    注释方法 PPOCR::benchmark_log 的内容。
// #include "auto_log/autolog.h"
...
void PaddleStructure::benchmark_log(int img_num) {
// if (this->time_info_det[0] + this->time_info_det[1] +
// this->time_info_det[2] >
// 0) {
// AutoLogger autolog_det("ocr_det", FLAGS_use_gpu, FLAGS_use_tensorrt,
// FLAGS_enable_mkldnn, FLAGS_cpu_threads, 1,
// "dynamic", FLAGS_precision, this->time_info_det,
// img_num);
// autolog_det.report();
// }
// if (this->time_info_rec[0] + this->time_info_rec[1] +
// this->time_info_rec[2] >
// 0) {
// AutoLogger autolog_rec("ocr_rec", FLAGS_use_gpu, FLAGS_use_tensorrt,
// FLAGS_enable_mkldnn, FLAGS_cpu_threads,
// FLAGS_rec_batch_num, "dynamic", FLAGS_precision,
// this->time_info_rec, img_num);
// autolog_rec.report();
// }
// if (this->time_info_cls[0] + this->time_info_cls[1] +
// this->time_info_cls[2] >
// 0) {
// AutoLogger autolog_cls("ocr_cls", FLAGS_use_gpu, FLAGS_use_tensorrt,
// FLAGS_enable_mkldnn, FLAGS_cpu_threads,
// FLAGS_cls_batch_num, "dynamic", FLAGS_precision,
// this->time_info_cls, img_num);
// autolog_cls.report();
// }
// if (this->time_info_table[0] + this->time_info_table[1] +
// this->time_info_table[2] >
// 0) {
// AutoLogger autolog_table("table", FLAGS_use_gpu, FLAGS_use_tensorrt,
// FLAGS_enable_mkldnn, FLAGS_cpu_threads,
// FLAGS_cls_batch_num, "dynamic", FLAGS_precision,
// this->time_info_table, img_num);
// autolog_table.report();
// }
// if (this->time_info_layout[0] + this->time_info_layout[1] +
// this->time_info_layout[2] >
// 0) {
// AutoLogger autolog_layout("layout", FLAGS_use_gpu, FLAGS_use_tensorrt,
// FLAGS_enable_mkldnn, FLAGS_cpu_threads,
// FLAGS_cls_batch_num, "dynamic",
// FLAGS_precision, this->time_info_layout,
// img_num);
// autolog_layout.report();
// }
}
...
  • 修改include/utility.cpp文件,内容如下:

    不使用 dirent.h

    注释行 // #include <dirent.h>

    注释方法 Utility::GetAllFiles 的内容。
// #include <dirent.h>
...
// list all files under a directory
void Utility::GetAllFiles(const char *dir_name,
std::vector<std::string> &all_inputs) {
// if (NULL == dir_name) {
// std::cout << " dir_name is null ! " << std::endl;
// return;
// }
// struct stat s;
// stat(dir_name, &s);
// if (!S_ISDIR(s.st_mode)) {
// std::cout << "dir_name is not a valid directory !" << std::endl;
// all_inputs.push_back(dir_name);
// return;
// } else {
// struct dirent *filename; // return value for readdir()
// DIR *dir; // return value for opendir()
// dir = opendir(dir_name);
// if (NULL == dir) {
// std::cout << "Can not open dir " << dir_name << std::endl;
// return;
// }
// std::cout << "Successfully opened the dir !" << std::endl;
// while ((filename = readdir(dir)) != NULL) {
// if (strcmp(filename->d_name, ".") == 0 ||
// strcmp(filename->d_name, "..") == 0)
// continue;
// // img_dir + std::string("/") + all_inputs[0];
// all_inputs.push_back(dir_name + std::string("/") +
// std::string(filename->d_name));
// }
// }
}
...

编写调用程序

  • 在项目中创建文件:paddle_util.h

    内容如下:
#pragma once
using namespace std; #include <include/paddleocr.h> class PaddleUtil {
public:
PaddleUtil(); public:
static PaddleUtil &get();
static void init();
void rec_image(const string &imageFile); private:
PaddleOCR::PPOCR ocr;
};
  • 在项目中创建文件:paddle_util.cpp

    内容如下:
#include "opencv2/core.hpp"
#include "opencv2/imgcodecs.hpp"
#include "opencv2/imgproc.hpp"
#include <iostream>
#include <vector> #include "paddle_util.h"
#include <include/args.h> using namespace PaddleOCR; PaddleUtil::PaddleUtil() {} PaddleUtil &PaddleUtil::get() {
static PaddleUtil self;
return self;
} void PaddleUtil::init() {
FLAGS_det = true;
FLAGS_rec = true;
FLAGS_cls = false;
FLAGS_use_angle_cls = false;
FLAGS_det_model_dir = "model/whl/det/en/en_PP-OCRv3_det_infer";
FLAGS_rec_model_dir = "model/whl/rec/en/en_PP-OCRv4_rec_infer";
FLAGS_rec_char_dict_path = "model/en_dict.txt";
} void PaddleUtil::rec_image(const string &imageFile) {
if (FLAGS_benchmark) {
ocr.reset_timer();
} cv::Mat img = cv::imread(imageFile, cv::IMREAD_COLOR);
if (!img.data) {
std::cerr << "[ERROR] image read failed! image path: " << imageFile
<< std::endl;
return;
} std::vector<OCRPredictResult> ocr_result =
ocr.ocr(img, FLAGS_det, FLAGS_rec, FLAGS_cls); Utility::print_result(ocr_result);
if (FLAGS_visualize && FLAGS_det) {
std::string file_name = Utility::basename(imageFile);
Utility::VisualizeBboxes(img, ocr_result, FLAGS_output + "/" + file_name);
}
}
  • 在项目中创建主程序文件:my-paddleocr.cpp

    内容如下:
#include <iostream>

#include "paddle_util.h"

int main() {

  PaddleUtil::init();
PaddleUtil::get().rec_image("model/254.jpg");
std::cout << "Done!\n";
}

编译运行

  • 在 Visual Studio 中使用 Release 版本编译。
  • 运行 copy_paddle_dll.bat,复制 paddle_inference.dll 到输出目录。
@REM copy dll to Release
copy /Y %PADDLE_ROOT%\paddle\lib\paddle_inference.dll .\x64\Release\
copy /Y %PADDLE_ROOT%\third_party\install\mkldnn\lib\mkldnn.dll .\x64\Release\
copy /Y %PADDLE_ROOT%\third_party\install\mklml\lib\mklml.dll .\x64\Release\
copy /Y %PADDLE_ROOT%\third_party\install\mklml\lib\libiomp5md.dll .\x64\Release\
copy /Y %PADDLE_ROOT%\third_party\install\paddle2onnx\lib\paddle2onnx.dll .\x64\Release\
copy /Y %PADDLE_ROOT%\third_party\install\onnxruntime\lib\onnxruntime.dll .\x64\Release\

在 C++ 项目中,通过源码使用 PaddlePaddle 实现 OCR 功能的更多相关文章

  1. centos7中通过源码安装postgresql13.6

    下载地址:https://www.postgresql.org/ftp/source/ 0.安装相关依赖库 centos依赖包下载地址:https://developer.aliyun.com/pac ...

  2. 通过源码了解ASP.NET MVC 几种Filter的执行过程 在Winform中菜单动态添加“最近使用文件”

    通过源码了解ASP.NET MVC 几种Filter的执行过程   一.前言 之前也阅读过MVC的源码,并了解过各个模块的运行原理和执行过程,但都没有形成文章(所以也忘得特别快),总感觉分析源码是大神 ...

  3. 通过源码分析Java开源任务调度框架Quartz的主要流程

    通过源码分析Java开源任务调度框架Quartz的主要流程 从使用效果.调用链路跟踪.E-R图.循环调度逻辑几个方面分析Quartz. github项目地址: https://github.com/t ...

  4. 通过源码了解ASP.NET MVC 几种Filter的执行过程

    一.前言 之前也阅读过MVC的源码,并了解过各个模块的运行原理和执行过程,但都没有形成文章(所以也忘得特别快),总感觉分析源码是大神的工作,而且很多人觉得平时根本不需要知道这些,会用就行了.其实阅读源 ...

  5. Linux下通过源码编译安装程序

    本文简单的记录了下,在linux下如何通过源码安装程序,以及相关的知识.(大神勿喷^_^) 一.程序的组成部分 Linux下程序大都是由以下几部分组成: 二进制文件:也就是可以运行的程序文件 库文件: ...

  6. 在centos6.7通过源码安装python3.6.7报错“zipimport.ZipImportError: can't decompress data; zlib not available”

    在centos6.7通过源码安装python3.6.7报错: zipimport.ZipImportError: can't decompress data; zlib not available 从 ...

  7. Kafka详解六:Kafka如何通过源码实现监控

    问题导读: 1.kafka的消费者组的消费偏移存储,kafka支持两个版本?        2.ConsumerOffsetChecker类的作用是什么?        3.Kafka如何通过源码实现 ...

  8. 通过源码编译安装VIM

    开发中使用的是Ubuntu 12.04 LTS,通过sudo apt-get install vim安装的版本较低,不支持YCM,所以,用源码编译并安装最新的Vim. 卸载旧版本的Vim: sudo ...

  9. echarts 通过源码方法 传入对应data数据获取分割步长值

    通过源码方法获取这里的分割数字长度 /** * Quantity of a number. e.g. 0.1, 1, 10, 100 * * @param {number} val * @return ...

  10. 通过源码安装PostgresSQL

    通过源码安装PostgresSQL 1.1 下载源码包环境: Centos6.8 64位 yum -y install bison flex readline-devel zlib-devel yum ...

随机推荐

  1. 【踩了一个坑】为什么 golang struct 中的 slice 无法原子赋值

    作者:张富春(ahfuzhang),转载时请注明作者和引用链接,谢谢! cnblogs博客 zhihu Github 公众号:一本正经的瞎扯 有这样一个结构体: type MySt struct{ F ...

  2. 发布.net core应用程序并部署到IIS上

    一.在项目里右击选择发布点击启动配置如下图所示 二.在打开的发布选项选择 配置 Release或DeBug ,目标框架选择对应的.net Core版本默认就行,部署模式有两种选择 1.框架依赖---- ...

  3. 复原docker中容器的启动命令

    复原 docker 容器的启动命令 前言 查看 docker 容器的启动命令 参考 复原 docker 容器的启动命令 前言 不规范的操作,在启动 docker 容器,没有留命令脚本,或者没有使用 d ...

  4. 设计模式学习-使用go实现责任链模式

    责任链模式 定义 优点 缺点 适用范围 代码实现 责任链模式对比装饰模式 参考 责任链模式 定义 责任链模式(Chain Of Responsibility):使多个对象都有机会处理请求,从而避免请求 ...

  5. C# - 将HTML网页、HTML字符串转换为PDF

    将HTML转换为PDF可实现格式保留.可靠打印.文档归档等多种用途,满足不同领域和情境下的需求.本文将通过以下两个示例,演示如何使用第三方库Spire.PDF for .NET和QT插件在C# 中将H ...

  6. 零基础入门学习Java之多线程

    多线程 话不多说,看代码 1.什么是多线程 众所周知CPU单线程的东西,也就是说在同一时间内程序只能去做一件事情,但很多时候比如说多人买票.龟兔赛跑.游戏开发等都需要在同一时间内完成多个东西,因此就有 ...

  7. 将实体光盘制作成光盘映像iso文件

    春节假期整理历史物件时发现一些书籍的光盘,虽然买了多年但一直没有看过,因为自己在用的电脑都没有光驱.正好老爸的电脑是带光驱的,想着趁过节把这些光盘的内容读取出来存在NAS上方便后续使用. 使用Ultr ...

  8. Navicat 15 for MySQL 破解【我亲测可用】

    1.去官网下载正版 https://www.navicat.com.cn/ 2.破解下载:https://files-cdn.cnblogs.com/files/del88/NavicatKeygen ...

  9. Flink-启动后无法访问WebUI界面(Flink1.16)

    问题描述 通过./bin/start-cluster.sh启动Flink程序,正常启动后无法通过浏览器访问web UI界面,http://192.168.80.133:8081. 问题原因 Flink ...

  10. 通过程序自动设置网卡的“internet共享”选项

    操作系统 : Windows 10_x64 [版本 10.0.19042.685] Windows下可以通过网卡共享进行上网,但是需要在网卡的属性里面进行设置,需要在视窗界面进行操作,不能实现自动化. ...