File size: 1,385 Bytes
68d7827 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 | {
"model_type": "DRGradingModel",
"task": "diabetic-retinopathy-grading",
"architecture": "convnext_tiny",
"backbone_timm_name": "convnext_tiny",
"in_channels": 4,
"image_size": 384,
"num_classes": 5,
"ordinal_head": "CORN",
"num_thresholds": 4,
"use_cbam": true,
"cbam_num_stages": 2,
"head_hidden_dim": 512,
"dropout": 0.4,
"channels_last": true,
"ema_decay": 0.999,
"loss_type": "corn",
"use_class_weighting": true,
"use_mixup": true,
"primary_weights": "*_best_ema.safetensors (EMA; reproduces paper metrics)",
"secondary_weights": "*_best.safetensors (raw model_state_dict)",
"normalization": {
"rgb": {
"mean": [
0.485,
0.456,
0.406
],
"std": [
0.229,
0.224,
0.225
]
},
"extra_channel": "Already in [0,1]; appended after RGB normalization; not ImageNet-normalized."
},
"class_names": [
"No DR",
"Mild",
"Moderate",
"Severe",
"Proliferative DR"
],
"extra_channel": {
"source": "segmentation_pooled_morph",
"producer": "gradeeye/seg-unet-bcedice",
"description": "The BCE+Dice U-Net raw sigmoid probability map after 3x3 elliptical morphological opening, 3x3 elliptical closing, and Gaussian blur (sigma=0.5). It remains a continuous [0,1] soft-probability map, not a morphological gradient or edge channel."
}
}
|