Marmelator/cv-analysis-service
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 37 Wiki Activity

better annotation output

Browse Source
This commit is contained in:
Matthias Bisping 2022-01-21 10:28:52 +01:00
parent d595b9a558
commit 2c9d895a91
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
table_parsing/table_parsig.py
View File

@ -1,3 +1,4 @@
import json
from itertools import count
import cv2
@ -32,7 +33,14 @@ def parse_tables_in_pdf(pages):
def annotate_image(image, stats):
for x, y, w, h, area in stats[2:]:
cv2.rectangle(image, (x, y), (x + w, y + h), (255, 0, 255), 2)
if w > 10 and h > 10:
cv2.rectangle(image, (x, y), (x + w, y + h), (255, 0, 255), 2)
for i, (s, v) in enumerate(zip(["x", "y", "w", "h"], [x, y, w, h])):
anno = f"{s} = {v}"
xann = int(x + 5)
yann = int(y + h - (20 * (i + 1)))
cv2.putText(image, anno, (xann, yann), cv2.FONT_HERSHEY_SIMPLEX, 0.7, (255, 0, 255), 2)
return image