mirror of
https://github.com/mikegerber/ocrd_calamari.git
synced 2025-06-09 19:59:53 +02:00
Merge pull request #90 from OCR-D/tf_disable_interactive_logs
disable tensorflow/keras print-to-stdout with tf_disable_interactive_…
This commit is contained in:
commit
c0a4dfde66
2 changed files with 16 additions and 9 deletions
|
@ -2,4 +2,3 @@ import json
|
||||||
from pkg_resources import resource_string
|
from pkg_resources import resource_string
|
||||||
|
|
||||||
OCRD_TOOL = json.loads(resource_string(__name__, 'ocrd-tool.json').decode('utf8'))
|
OCRD_TOOL = json.loads(resource_string(__name__, 'ocrd-tool.json').decode('utf8'))
|
||||||
TF_CPP_MIN_LOG_LEVEL = '3' # '3' == ERROR
|
|
||||||
|
|
|
@ -5,6 +5,21 @@ import itertools
|
||||||
from glob import glob
|
from glob import glob
|
||||||
|
|
||||||
import numpy as np
|
import numpy as np
|
||||||
|
from ocrd_utils import (
|
||||||
|
assert_file_grp_cardinality,
|
||||||
|
concat_padded,
|
||||||
|
coordinates_for_segment,
|
||||||
|
getLogger,
|
||||||
|
make_file_id,
|
||||||
|
points_from_polygon,
|
||||||
|
polygon_from_x0y0x1y1,
|
||||||
|
MIMETYPE_PAGE,
|
||||||
|
tf_disable_interactive_logs,
|
||||||
|
)
|
||||||
|
|
||||||
|
# Disable tensorflow/keras logging via print before importing calamari
|
||||||
|
tf_disable_interactive_logs()
|
||||||
|
|
||||||
from calamari_ocr import __version__ as calamari_version
|
from calamari_ocr import __version__ as calamari_version
|
||||||
from calamari_ocr.ocr import MultiPredictor
|
from calamari_ocr.ocr import MultiPredictor
|
||||||
from calamari_ocr.ocr.voting import voter_from_proto
|
from calamari_ocr.ocr.voting import voter_from_proto
|
||||||
|
@ -18,16 +33,9 @@ from ocrd_models.ocrd_page import (
|
||||||
WordType, GlyphType, CoordsType,
|
WordType, GlyphType, CoordsType,
|
||||||
to_xml
|
to_xml
|
||||||
)
|
)
|
||||||
from ocrd_utils import (
|
|
||||||
getLogger, concat_padded,
|
|
||||||
coordinates_for_segment, points_from_polygon, polygon_from_x0y0x1y1,
|
|
||||||
make_file_id, assert_file_grp_cardinality,
|
|
||||||
MIMETYPE_PAGE
|
|
||||||
)
|
|
||||||
|
|
||||||
from ocrd_calamari.config import OCRD_TOOL, TF_CPP_MIN_LOG_LEVEL
|
from ocrd_calamari.config import OCRD_TOOL
|
||||||
|
|
||||||
os.environ['TF_CPP_MIN_LOG_LEVEL'] = TF_CPP_MIN_LOG_LEVEL
|
|
||||||
from tensorflow import __version__ as tensorflow_version
|
from tensorflow import __version__ as tensorflow_version
|
||||||
|
|
||||||
TOOL = 'ocrd-calamari-recognize'
|
TOOL = 'ocrd-calamari-recognize'
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue