ayyuce commited on
Commit
d200158
·
verified ·
1 Parent(s): bdd6257

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +34 -0
README.md ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ datasets:
3
+ - RationAI/PanNuke
4
+ tags:
5
+ - medical
6
+ - pannuke
7
+ - dinov2
8
+ - segmentation
9
+ - nuclei-segmentation
10
+ license: gpl-3.0
11
+ pipeline_tag: image-segmentation
12
+ ---
13
+ # DINOv2 Nuclei Segmentation Model
14
+
15
+ ## Model Details
16
+ - **Format**: PyTorch .pth checkpoint
17
+ - **Architecture**: DINOv2-Base (vitb14) with custom decoder
18
+ - **Training Data**: PanNuke dataset
19
+ - **Input Size**: 252x252
20
+ - **Classes**: Background, Neoplastic, Inflammatory, Connective, Dead, Epithelial
21
+
22
+ ## Usage
23
+ ```python
24
+ from huggingface_hub import hf_hub_download
25
+
26
+ model = DinoV2ForSegmentation(num_classes=6)
27
+ model.load_state_dict(
28
+ torch.load(
29
+ hf_hub_download(
30
+ repo_id="{model_repo}",
31
+ filename="{best_checkpoint}"
32
+ )
33
+ )
34
+ )