README.md

---
datasets:
- RationAI/PanNuke
tags:
- medical
- pannuke
- dinov2
- segmentation
- nuclei-segmentation
license: gpl-3.0
pipeline_tag: image-segmentation
---
# DINOv2 Nuclei Segmentation Model

## Model Details
- **Format**: PyTorch .pth checkpoint
- **Architecture**: DINOv2-Base (vitb14) with custom decoder
- **Training Data**: PanNuke dataset
- **Input Size**: 252x252
- **Classes**: Background, Neoplastic, Inflammatory, Connective, Dead, Epithelial

## Usage
```python
from huggingface_hub import hf_hub_download

model = DinoV2ForSegmentation(num_classes=6)
model.load_state_dict(
    torch.load(
        hf_hub_download(
            repo_id="{model_repo}",
            filename="{best_checkpoint}"
        )
    )
)