Echo/fcb_photo_review
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

修正文字分割时的方向问题

Browse Source
This commit is contained in:
liuyebo
2024-07-10 14:58:26 +08:00
parent 312f35f832
commit c11c0c0c6e
1 changed files with 54 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
photo_mask.py
View File

@@ -184,32 +184,72 @@ def get_ocr_layout(ocr, img_path):
def find_box_of_content(content, layout): def find_box_of_content(content, layout):
full_box = layout[0] full_box = layout[0]
box_len = full_box[2] - full_box[0] x_len = full_box[2] - full_box[0]
y_len = full_box[3] - full_box[1]
if x_len >= y_len:
# 横向排布
box_len = x_len
direction = "x"
else:
# 纵向排布
box_len = y_len
direction = "y"
text = layout[1] text = layout[1]
text_len = len(text) text_len = len(text)
char_len = box_len / text_len char_len = box_len / text_len
index = text.index(content) index = text.index(content)
if direction == "x":
# 横向排布
return ( return (
full_box[0] + index * char_len, full_box[0] + index * char_len,
full_box[1], full_box[1],
full_box[0] + (index + len(content) + 1) * char_len, full_box[0] + (index + len(content) + 1) * char_len,
full_box[3], full_box[3],
) )
else:
# 纵向排布
return (
full_box[0],
full_box[1] + index * char_len,
full_box[2],
full_box[1] + (index + len(content) + 1) * char_len,
)
def find_box_of_value(key, layout): def find_box_of_value(key, layout):
full_box = layout[0] full_box = layout[0]
box_len = full_box[2] - full_box[0] x_len = full_box[2] - full_box[0]
y_len = full_box[3] - full_box[1]
if x_len >= y_len:
# 横向排布
box_len = x_len
direction = "x"
else:
# 纵向排布
box_len = y_len
direction = "y"
text = layout[1] text = layout[1]
text_len = len(text) text_len = len(text)
char_len = box_len / text_len char_len = box_len / text_len
index = text.index(key) index = text.index(key)
if direction == "x":
# 横向排布
return ( return (
full_box[0] + (index + len(key)) * char_len, full_box[0] + (index + len(key)) * char_len,
full_box[1], full_box[1],
full_box[0] + (index + len(key) + 4) * char_len, full_box[0] + (index + len(key) + 4) * char_len,
full_box[3], full_box[3],
) )
else:
# 纵向排布
return (
full_box[0],
full_box[1] + (index + len(key)) * char_len,
full_box[2],
full_box[1] + (index + len(key) + 4) * char_len,
)
def get_mask_layout(image, content): def get_mask_layout(image, content):