File size: 325 Bytes
bdd6257
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
config = {
    "model_type": "dinov2-segmentation",
    "checkpoint_name": best_checkpoint,
    "num_classes": 6,
    "image_size": 252,
    "class_names": [
        "Background", "Neoplastic", "Inflammatory",
        "Connective", "Dead", "Epithelial"
    ],
    "architecture": "DINOv2-Base with custom segmentation head"
}