50 lines
1.6 KiB
Python
50 lines
1.6 KiB
Python
|
# 排版解析-单栏-自然段
|
||
|
|
|
||
|
|
from .parser_single_line import SingleLine
|
||
|
|
from .parser_tools.line_preprocessing import linePreprocessing # 行预处理
|
||
|
|
from .parser_tools.paragraph_parse import ParagraphParse # 段内分析器
|
||
|
|
|
||
|
|
|
||
|
|
class SinglePara(SingleLine):
|
||
|
|
def __init__(self):
|
||
|
|
self.tbpuName = "排版解析-单栏-自然段"
|
||
|
|
|
||
|
|
# 段内分析器对象
|
||
|
|
get_info = lambda tb: (tb["normalized_bbox"], tb["text"])
|
||
|
|
|
||
|
|
def set_end(tb, end): # 获取预测的块尾分隔符
|
||
|
|
tb["line"][-1]["end"] = end
|
||
|
|
|
||
|
|
self.pp = ParagraphParse(get_info, set_end)
|
||
|
|
|
||
|
|
def run(self, textBlocks):
|
||
|
|
textBlocks = linePreprocessing(textBlocks) # 预处理
|
||
|
|
lines = self.get_lines(textBlocks) # 获取每一行
|
||
|
|
# 将行封装为tb
|
||
|
|
temp_tbs = []
|
||
|
|
for line in lines:
|
||
|
|
b0, b1, b2, b3 = line[0]["normalized_bbox"]
|
||
|
|
# 搜索bbox
|
||
|
|
for i in range(1, len(line)):
|
||
|
|
bb = line[i]["normalized_bbox"]
|
||
|
|
b1 = min(b1, bb[1])
|
||
|
|
b2 = max(b1, bb[2])
|
||
|
|
b3 = max(b1, bb[3])
|
||
|
|
# 构建tb
|
||
|
|
temp_tbs.append(
|
||
|
|
{
|
||
|
|
"normalized_bbox": (b0, b1, b2, b3),
|
||
|
|
"text": line[0]["text"][0] + line[-1]["text"][-1],
|
||
|
|
"line": line,
|
||
|
|
}
|
||
|
|
)
|
||
|
|
# 预测结尾分隔符
|
||
|
|
self.pp.run(temp_tbs)
|
||
|
|
# 解包
|
||
|
|
textBlocks = []
|
||
|
|
for t in temp_tbs:
|
||
|
|
for tb in t["line"]:
|
||
|
|
del tb["normalized_bbox"]
|
||
|
|
textBlocks.append(tb)
|
||
|
|
return textBlocks
|