microsoft

/

table-transformer-structure-recognition-v1.1-all

like 56

Follow

Microsoft 4,773

How is it possble, that if I insert this perfectly cropped out table:

I get this bad result:

But if there is still a bit of uncropped text like in:

I get much, much better results:

I need to be able to work with perfect cropped out tables, because I have a good DETR model for extracting these...
Used Code:
from transformers import TableTransformerForObjectDetection
model = TableTransformerForObjectDetection.from_pretrained("microsoft/table-transformer-structure-recognition")

def cell_detection(file_path):

image = Image.open(file_path).convert("RGB")
print()
width, height = image.size
image.resize((int(width*0.5), int(height*0.5)))


encoding = feature_extractor(image, return_tensors="pt")
encoding.keys()

with torch.no_grad():
  outputs = model(**encoding)


target_sizes = [image.size[::-1]]
results = feature_extractor.post_process_object_detection(outputs, threshold=0.6, target_sizes=target_sizes)[0]
plot_results(image, results['scores'], results['labels'], results['boxes'])
model.config.id2label

file_path = r"C:\Users\blur\Coding\blur\Segmentation\test_images\perfect_table_inkltext.png"
#file_path = r"C:\Users\blur\Coding\blur\Segmentation\test_images\perfect_table.png"
#file_path = r"C:\Users\blur\Coding\blur\Segmentation\test_images\test_kaggle.png"
cell_detection(file_path)