qurator-spk/page2tsv
1
0
Fork 0
mirror of https://github.com/qurator-spk/page2tsv.git synced 2026-03-15 19:52:04 +01:00
Code Issues Releases Wiki Activity

install into qurator namespace

Browse source
This commit is contained in:
Konstantin Baierer 2022-11-08 16:19:23 +01:00
parent abeca0df16
commit e1a440b91c
10 changed files with 17 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

23
tsvtools/ocr.py
View file

@ -1,23 +0,0 @@
import numpy as np
import pandas as pd
def get_conf_color(conf, min_conf, max_conf):
conf = min_conf if conf < min_conf else conf
conf = max_conf if conf > max_conf else conf
interval_size = (max_conf - min_conf) / 2.0
colors = np.array([[216, 108, 117], [216, 206, 108], [108, 216, 146]])
colors = pd.DataFrame(colors, index=[0, 1, 2], columns=['R', 'G', 'B'])
lower = np.floor((conf - min_conf) / interval_size)
upper = np.ceil((conf - min_conf) / interval_size)
pos = (conf - min_conf) / (2.0*interval_size)
col = (colors.loc[lower] * (1.0 - pos) + colors.loc[upper] * pos).astype(int)
return '#{:02x}'.format(col.R) + '{:02x}'.format(col.G) + '{:02x}'.format(col.B)