DataCatalogue is a research project jointly led by Inria, the Bibliothèque nationale de France (National Library of France), and the Institut national d'histoire de l'art (National Institute of Art History).
It aims at restructuring OCR-ed auction sale catalogs kept in France national collections into TEI-XML, using machine learning solutions.
We trained a YOLOv5 model on custom data to perform document layout analysis on auction sale catalogs.
The training set consists of 581 images, annotated with two classes:
- title (585 instances)
- entry (it refers to a catalog entry) (5017 instances)
59 images were used for validation.
The dataset is not released for the moment.
An interactive demo is available on the following HugginFace Space: https://huggingface.co/spaces/HugoSchtr/DataCat_Yolov5
The model performs well on our data and now needs to be incorporated into a dedicated pipeline for the research project.
We also plan to train a new model on a larger training set in the near future.
- Downloads last month