| 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" | |
| } |