Downloads · 30 days
0
yolov5_datacat
HugoSchtr/yolov5_datacat
yolov5_datacat is a machine learning model from HugoSchtr. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as cc-by-4.0.
YOLOv5 is an open-source object detection model released by Ultralytics, on Github.
Access
Public
Updated Dec 20, 2022
Repo size
176 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pt173 MB · 99%
At a glance
- License
- cc-by-4.0
- Access
- Public
- Created
- Dec 19, 2022
- Updated
- Dec 20, 2022
- SHA
- 8824445c
Datasets
Spaces
Model card
From the Hugging Face model README
What's YOLOv5
YOLOv5 is an open-source object detection model released by Ultralytics, on Github.
DataCatalogue (or DataCat)
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.
DataCat Yolov5
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.
We reached:
| precision | recall | mAP_0.5 | mAP_0.5:0.95 |
|---|---|---|---|
| 0.99 | 0.99 | 0.98 | 0.75 |
Dataset
The dataset is not released for the moment.
Demo
An interactive demo is available on the following HugginFace Space: https://huggingface.co/spaces/HugoSchtr/DataCat_Yolov5
<img alt='detection example' src="https://huggingface.co/HugoSchtr/yolov5_datacat/resolve/main/eval/detection_example.png" width=30% height=30%>What's next
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.
- License
- cc-by-4.0
- Created
- Dec 19, 2022
- Updated
- Dec 20, 2022