Downloads · 30 days
21
16% of all-time downloads
dinov2-nuclei-segmentation
ayyuce/dinov2-nuclei-segmentation
dinov2-nuclei-segmentation is a image segmentation model from ayyuce. Use it for the image segmentation task on the model card, and read the license before you ship it in a product. The card lists the license as gpl-3.0.
- Format: PyTorch .pth checkpoint - Architecture: DINOv2-Base (vitb14) with custom decoder - Training Data: PanNuke dataset - Input Size: 252x252 - Classes: Background, Neoplastic, Inflammatory, Connective, Dead, Epit…
All-time downloads
134
Public
Repo size
355 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pth355 MB · 100%
At a glance
- Task
- Image Segmentation
- License
- gpl-3.0
- Access
- Public
- Created
- Mar 20, 2025
- Updated
- Mar 20, 2025
- SHA
- d2001585
Datasets
Model card
From the Hugging Face model README
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
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}"
)
)
)
- Task
- Image Segmentation
- License
- gpl-3.0
- Created
- Mar 20, 2025
- Updated
- Mar 20, 2025