--- 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.