|
--- |
|
library_name: YOLO |
|
license: cc0-1.0 |
|
tags: |
|
- YOLO |
|
- object-detection |
|
- dla |
|
- historical |
|
metrics: |
|
- P |
|
- R |
|
- mAP@50 |
|
- mAP@50-95 |
|
pipeline_tag: object-detection |
|
version: |
|
- YOLOv8 |
|
--- |
|
|
|
# YOLOv8 - PELLET Casimir Marius - Line + Illustration detection |
|
|
|
This model predicts the following elements from document images: |
|
- text line |
|
- illustration |
|
|
|
## Model description |
|
|
|
The model has been trained using the YOLOv8 library on [PELLET Casimir Marius](https://huggingface.co./datasets/Teklia/PELLET-Casimir-Marius-yolov8) dataset. |
|
It has been trained on images with their dimensions equal to 640 pixels, starting from the YOLOv8m checkpoint. |
|
|
|
## Evaluation results |
|
|
|
The model achieves the following results: |
|
|
|
| Set | Class | Images | Instances | P | R | mAP@50 | mAP@50-95 | |
|
| ----- | ------------ | ------ | --------- | ----- | ----- | ------ | --------- | |
|
| train | all | 80 | 881 | 0.707 | 0.823 | 0.825 | 0.457 | |
|
| | text line | 80 | 843 | 0.801 | 0.73 | 0.781 | 0.275 | |
|
| | illustration | 80 | 38 | 0.613 | 0.916 | 0.87 | 0.64 | |
|
| val | all | 10 | 129 | 0.608 | 0.872 | 0.847 | 0.493 | |
|
| | text line | 10 | 125 | 0.791 | 0.744 | 0.7 | 0.256 | |
|
| | illustration | 10 | 4 | 0.424 | 1 | 0.995 | 0.731 | |
|
| test | all | 10 | 125 | 0.767 | 0.881 | 0.932 | 0.556 | |
|
| | text line | 10 | 122 | 0.903 | 0.763 | 0.868 | 0.331 | |
|
| | illustration | 10 | 3 | 0.631 | 1 | 0.995 | 0.78 | |
|
|
|
## How to use? |
|
|
|
- Download the [weights of this model](https://huggingface.co./Teklia/PELLET-CasimirMarius-YOLOv8/resolve/main/model.pt?download=true); |
|
- Refer to the [Ultralytics documentation](https://docs.ultralytics.com/modes/predict/) to use this model. |
|
|
|
|
