| | ---
|
| | license: cc-by-nc-3.0
|
| | pipeline_tag: image-to-image
|
| | tags:
|
| | - vision
|
| | - document-processing
|
| | - binarization
|
| | - segmentation
|
| | ---
|
| |
|
| | # Tzefa Binarization Model (mit_b5 HighResMAnet)
|
| |
|
| | Custom-trained document binarization model for the Tzefa OCR pipeline.
|
| |
|
| | ## Architecture
|
| | - **Encoder:** MiT-B5 (Mix Transformer)
|
| | - **Decoder:** MAnet with custom High-Resolution Stem + Fusion Head
|
| | - **Framework:** segmentation-models-pytorch
|
| | - **Input:** RGB image tiles (640x640)
|
| | - **Output:** Binary mask (ink=0, paper=255)
|
| |
|
| | ## Usage
|
| | ```python
|
| | from huggingface_hub import hf_hub_download
|
| | import torch
|
| |
|
| | # Download weights
|
| | ckpt_path = hf_hub_download("WARAJA/b5_model", "b5_model.pth")
|
| |
|
| | # Load model (see Tzefa Binarization Space for full architecture code)
|
| | checkpoint = torch.load(ckpt_path, map_location="cpu")
|
| | ```
|
| |
|
| | ## Related
|
| | - [Binarization Demo](https://huggingface.co/spaces/WARAJA/Tzefa-Binarization)
|
| | - [Full Tzefa Pipeline](https://huggingface.co/spaces/WARAJA/Tzefa)
|
| | - [Binarization Dataset](https://huggingface.co/datasets/WARAJA/Tzefa-Binarization-Dataset)
|
| | |
