Spaces:
Sleeping
Sleeping
File size: 1,026 Bytes
9e9a60d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 | from table2html.source import visualize_boxes
from table2html.source import crop_image
from table2html import Table2HTML
import cv2
table_crop_padding = 15
table_config = {
}
row_config = {
}
column_config = {
}
table2html = Table2HTML(table_config, row_config, column_config)
image = cv2.imread(
r"C:\Users\jayll\Downloads\no_tables.jpg")
detection_data = table2html(image, table_crop_padding)
for i, data in enumerate(detection_data):
table_image = crop_image(image, data["table_bbox"], table_crop_padding)
cv2.imwrite(
"table_detection.jpg",
visualize_boxes(
image,
[data["table_bbox"]],
color=(0, 0, 255),
thickness=1
)
)
cv2.imwrite(
"structure_detection.jpg",
visualize_boxes(
table_image,
[cell['box'] for cell in data['cells']],
color=(0, 255, 0),
thickness=1
)
)
with open(f"table_{i}.html", "w") as f:
f.write(data["html"]) |