Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Refactor recognize #25

Merged
merged 5 commits into from
Oct 26, 2018
Merged

Refactor recognize #25

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
06c9cab
recognize: clean up imports and pylint warns
kba Oct 25, 2018
6ff10e1
split recognize.process into smaller chunks
kba Oct 25, 2018
a7c0da2
smoke test cli
kba Oct 25, 2018
cc6b563
move locale setting before tesserocr import, debug output for cli test
kba Oct 25, 2018
40fa850
split recognizing existing glyphs vs. all in word
kba Oct 25, 2018
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions .pylintrc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,8 @@ ignored-modules=cv2,tesserocr
[MESSAGES CONTROL]
disable =
ungrouped-imports,
fixme,
bad-continuation,
missing-docstring,
no-self-use,
too-many-arguments,
Expand Down
10 changes: 6 additions & 4 deletions ocrd_tesserocr/recognize.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,15 +1,16 @@
from __future__ import absolute_import

import locale
locale.setlocale(locale.LC_ALL, 'C') # circumvent tesseract-ocr issue 1670 (which cannot be done on command line because Click requires an UTF-8 locale in Python 3)

from tesserocr import RIL, PSM, PyTessBaseAPI, PyResultIterator, get_languages, iterate_level
from ocrd.utils import getLogger, concat_padded, xywh_from_points, points_from_xywh, points_from_x0y0x1y1
from tesserocr import RIL, PSM, PyTessBaseAPI, get_languages
from ocrd.utils import getLogger, concat_padded, xywh_from_points, points_from_x0y0x1y1
from ocrd.model.ocrd_page import from_file, to_xml, TextEquivType, CoordsType, GlyphType, WordType
from ocrd.model.ocrd_page_generateds import TextStyleType, MetadataItemType, LabelsType, LabelType
from ocrd import Processor, MIMETYPE_PAGE
from ocrd_tesserocr.config import TESSDATA_PREFIX, OCRD_TOOL

locale.setlocale(locale.LC_ALL, 'C') # circumvent tesseract-ocr issue 1670 (which cannot be done on command line because Click requires an UTF-8 locale in Python 3)
kba marked this conversation as resolved.
Show resolved Hide resolved

log = getLogger('processor.TesserocrRecognize')

CHOICE_THRESHOLD_NUM = 6 # maximum number of choices to query and annotate
Expand All @@ -28,6 +29,7 @@ def process(self):
Performs the (text) recognition.
"""
# print(self.parameter)
log.debug("TESSDATA: %s, installed tesseract models: %s", *get_languages())
maxlevel = self.parameter['textequiv_level']
model = get_languages()[1][-1] # last installed model
if 'model' in self.parameter:
Expand All @@ -38,7 +40,7 @@ def process(self):
log.info("Using model '%s' in %s for recognition at the %s level", model, get_languages()[0], maxlevel)
tessapi.SetVariable("glyph_confidences", "2") # populate GetChoiceIterator() with LSTM models, too
# tessapi.SetVariable("tessedit_single_match", "0")
#
#
# tessedit_load_sublangs
# tessedit_preserve_min_wd_len 2
# tessedit_prefer_joined_punct 0
Expand Down