Odia Ocr Qwen Finetuned v2

Training Notes

Training was early stopped at step 6,400 (of 12,387 planned) due to confirmed loss plateau:

• Train loss converged to ~4.83–5.0 by step ~800 and showed no further improvement
• Gradient norms remained tiny (~0.014–0.024) indicating saturated word-level learning
• Eval loss plateau: 5.512 → 5.454 (only 1% delta across 6,000 steps)

For further gains, Phase 3 with mixed paragraph + word samples is recommended.

Sample Predictions

Each row shows the original crop image from shantipriya/odia-ocr-merged, the ground truth label, the model-extracted text, and a quality remark.

✅ Good — clean, high-contrast printed crops

Majority case (~65–70%) for well-segmented printed word crops.

⚠️ Mixed — partial errors, diacritic / conjunct substitutions

Mixed cases (~20–25%) mostly involve complex conjuncts and long-vowel matras.

❌ Bad — degraded, truncated, or low-resolution outputs

Bad cases (~10–15%): very low resolution (<20 px height), heavy degradation, or long compound words.

Summary

Note: CER/WER metrics on a curated test split are pending. Percentages are estimated from qualitative review of ~200 samples.

Usage

from transformers import AutoProcessor, Qwen2_5_VLForConditionalGeneration
from peft import PeftModel
import torch
from PIL import Image

base_model    = "Qwen/Qwen2.5-VL-3B-Instruct"
adapter_model = "shantipriya/odia-ocr-qwen-finetuned_v2"

processor = AutoProcessor.from_pretrained(base_model, trust_remote_code=True)
model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
    base_model, torch_dtype=torch.float16, device_map="auto", trust_remote_code=True
)
model = PeftModel.from_pretrained(model, adapter_model)
model.eval()

def ocr_image(image_path: str) -> str:
    image = Image.open(image_path).convert("RGB")
    messages = [{
        "role": "user",
        "content": [
            {"type": "image", "image": image},
            {"type": "text",  "text": "Extract the Odia text from this image. Return only the text."}
        ]
    }]
    text_prompt = processor.apply_chat_template(
        messages, tokenize=False, add_generation_prompt=True
    )
    inputs = processor(text=[text_prompt], images=[image], return_tensors="pt").to(model.device)
    with torch.no_grad():
        output_ids = model.generate(**inputs, max_new_tokens=128, do_sample=False, temperature=1.0)
    generated = output_ids[:, inputs["input_ids"].shape[1]:]
    return processor.batch_decode(generated, skip_special_tokens=True)[0].strip()

print(ocr_image("odia_word.png"))

Training Data

The model was trained on shantipriya/odia-ocr-merged:

• 145,000 word-level Odia script image crops
• Diverse fonts, sizes, and print quality
• Sourced from multiple Odia OCR corpora and merged/deduplicated

Available Checkpoints

Limitations

• Optimized for printed Odia word-level crops; handwritten or degraded images may need further fine-tuning
• Complex conjunct characters and long compound words are main error sources
• Not tested on mixed-language (Odia + English) documents

Citation

@misc{parida2026odiaocr,
  author       = {Shantipriya Parida and OdiaGenAI Team},
  title        = {Odia OCR: Fine-tuned Qwen2.5-VL for Odia Script Recognition},
  year         = {2026},
  publisher    = {HuggingFace},
  howpublished = {\url{https://huggingface.co/shantipriya/odia-ocr-qwen-finetuned_v2}},
  note         = {LoRA fine-tune of Qwen2.5-VL-3B-Instruct on 145K Odia OCR word crops}
}

If using the training dataset, also cite:

@misc{parida2026odiadataset,
  author       = {Shantipriya Parida},
  title        = {Odia OCR Merged Dataset},
  year         = {2026},
  publisher    = {HuggingFace},
  howpublished = {\url{https://huggingface.co/datasets/shantipriya/odia-ocr-merged}}
}

License

Apache 2.0

Contact

• Organization: OdiaGenAI
• Mirror: OdiaGenAIOCR/odia-ocr-qwen-finetuned

checkpoint-5400/adapter_model.safetensors

safetensors · 1.11 GB · SHA-256 5b9f5b5d6bd7…63da · Hugging Face