trocr-base-handwritten-finetuned
TrOCR fine-tuné pour la reconnaissance d'écriture manuscrite, à partir de microsoft/trocr-base-stage1.
Résultats sur le jeu de test (778 exemples)
| Métrique | Valeur |
|---|---|
| CER | 50.15 % |
| WER | 94.96 % |
| Exact match | 0.00 % |
Utilisation
from transformers import VisionEncoderDecoderModel, TrOCRProcessor
from PIL import Image
processor = TrOCRProcessor.from_pretrained("dodo1234/trocr-base-handwritten-finetuned")
model = VisionEncoderDecoderModel.from_pretrained("dodo1234/trocr-base-handwritten-finetuned")
image = Image.open("ligne_manuscrite.jpg").convert("RGB")
pixel_values = processor(image, return_tensors="pt").pixel_values
ids = model.generate(pixel_values)
print(processor.batch_decode(ids, skip_special_tokens=True)[0])
Entraînement
- Base : microsoft/trocr-base-stage1
- Batch size : 8, fp16, Seq2SeqTrainer