qurator-spk/dinglehopper
1
0
Fork 0
mirror of https://github.com/qurator-spk/dinglehopper.git synced 2025-06-09 03:40:12 +02:00
Code Issues Releases Wiki Activity

🎨 dinglehopper: s/LOG/log

Browse source
This commit is contained in:
Gerber, Mike 2020-11-09 16:55:43 +01:00
parent 0804b029c4
commit 31c63f9e4c
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
qurator/dinglehopper/extracted_text.py
View file

@ -243,7 +243,7 @@ def get_textequiv_unicode(text_segment, nsmap) -> str:
def get_first_textequiv(textequivs, segment_id):
"""Get the first TextEquiv based on index or conf order if index is not present."""
LOG = getLogger('processor.OcrdDinglehopperEvaluate')
log = getLogger('processor.OcrdDinglehopperEvaluate')
if len(textequivs) == 1:
return textequivs[0]
@ -252,18 +252,18 @@ def get_first_textequiv(textequivs, segment_id):
nan_mask = np.isnan(indices)
if np.any(~nan_mask):
if np.any(nan_mask):
LOG.warning("TextEquiv without index in %s.", segment_id)
log.warning("TextEquiv without index in %s.", segment_id)
index = np.nanargmin(indices)
else:
# try ordering by conf
confidences = np.array([get_attr(te, 'conf') for te in textequivs], dtype=float)
if np.any(~np.isnan(confidences)):
LOG.info("No index attributes, use 'conf' attribute to sort TextEquiv in %s.",
log.info("No index attributes, use 'conf' attribute to sort TextEquiv in %s.",
segment_id)
index = np.nanargmax(confidences)
else:
# fallback to first entry in case of neither index or conf present
LOG.warning("No index attributes, use first TextEquiv in %s.", segment_id)
log.warning("No index attributes, use first TextEquiv in %s.", segment_id)
index = 0
return textequivs[index]