Docling IBM models

AI modules to support the Docling PDF document conversion project.

• TableFormer is an AI module that recognizes the structure of a table and the bounding boxes of the table content.

Install

The package provides two variants which allow to seemlessly switch between opencv-python and opencv-python-headless.

# Option 1: with opencv-python-headless
pip install "docling-ibm-models[opencv-python-headless]"

# Option 2: with opencv-python
pip install "docling-ibm-models[opencv-python]"

Pipeline Overview

Datasets

Below we list datasets used with their description, source, and "TableFormer Format". The TableFormer Format is our processed version of the version of the original format to work with the dataloader out of the box, and to augment the dataset when necassary to add missing groundtruth (bounding boxes for empty cells).

Models

TableModel04:

TableModel04rs (OTSL) is our SOTA method that using transformers in order to predict table structure and bounding box.

Configuration file

Example configuration can be found inside test tests/test_tf_predictor.py These are the main sections of the configuration file:

• dataset: The directory for prepared data and the parameters used during the data loading.
• model: The type, name and hyperparameters of the model. Also the directory to save/load the trained checkpoint files.
• train: Parameters for the training of the model.
• predict: Parameters for the evaluation of the model.
• dataset_wordmap: Very important part that contains token maps.

Model weights

You can download the model weights and config files from the links:

• TableFormer Checkpoint

Inference Tests

You can run the inference tests for the models with:

python -m pytest tests/

This will also generate prediction and matching visualizations that can be found here: tests\test_data\viz\

Visualization outlines:

• Light Pink: border of recognized table
• Grey: OCR cells
• Green: prediction bboxes
• Red: OCR cells matched with prediction
• Blue: Post processed, match
• Bold Blue: column header
• Bold Magenta: row header
• Bold Brown: section row (if table have one)