aji/ScreenTranslator
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Explicitly set page segmentation mode to auto

Browse Source
This commit is contained in:
Gres 2021-03-21 23:05:32 +03:00
parent efee1088f7
commit 5b591fa476
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/ocr/tesseract.cpp
View File

@ -137,6 +137,7 @@ class Tesseract::Wrapper
using GetUtf8 = char *(*)(TessBaseAPI *); using GetUtf8 = char *(*)(TessBaseAPI *);
using ClearApi = void (*)(TessBaseAPI *); using ClearApi = void (*)(TessBaseAPI *);
using DeleteUtf8 = void (*)(const char *); using DeleteUtf8 = void (*)(const char *);
using SetPageMode = void (*)(TessBaseAPI *, int);
public: public:
explicit Wrapper(const QString &libraryName) explicit Wrapper(const QString &libraryName)
@ -156,6 +157,8 @@ public:
ok &= bool(getUtf8_ = (GetUtf8)lib.resolve("TessBaseAPIGetUTF8Text")); ok &= bool(getUtf8_ = (GetUtf8)lib.resolve("TessBaseAPIGetUTF8Text"));
ok &= bool(clearApi_ = (ClearApi)lib.resolve("TessBaseAPIClear")); ok &= bool(clearApi_ = (ClearApi)lib.resolve("TessBaseAPIClear"));
ok &= bool(deleteUtf8_ = (DeleteUtf8)lib.resolve("TessDeleteText")); ok &= bool(deleteUtf8_ = (DeleteUtf8)lib.resolve("TessDeleteText"));
ok &= bool(setPageMode_ =
(SetPageMode)lib.resolve("TessBaseAPISetPageSegMode"));
if (!ok) { if (!ok) {
LERROR() << "Failed to resolve tesseract functions from" << libraryName; LERROR() << "Failed to resolve tesseract functions from" << libraryName;
return; return;
@ -184,6 +187,9 @@ public:
{ {
SOFT_ASSERT(handle_, return {}); SOFT_ASSERT(handle_, return {});
SOFT_ASSERT(setPageMode_, return {});
setPageMode_(handle_, 3); // PSM_AUTO
SOFT_ASSERT(setImage_, return {}); SOFT_ASSERT(setImage_, return {});
setImage_(handle_, pix); setImage_(handle_, pix);
LTRACE() << "Set Pix to engine"; LTRACE() << "Set Pix to engine";
@ -216,6 +222,7 @@ private:
GetUtf8 getUtf8_{nullptr}; GetUtf8 getUtf8_{nullptr};
ClearApi clearApi_{nullptr}; ClearApi clearApi_{nullptr};
DeleteUtf8 deleteUtf8_{nullptr}; DeleteUtf8 deleteUtf8_{nullptr};
SetPageMode setPageMode_{nullptr};
TessBaseAPI *handle_{nullptr}; TessBaseAPI *handle_{nullptr};
}; };