File size: 1,019 Bytes
3ff0b50 | 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 | {
"model_type": "DRGradingModel",
"task": "diabetic-retinopathy-grading",
"architecture": "convnext_tiny",
"backbone_timm_name": "convnext_tiny",
"in_channels": 3,
"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": "lodo_ddr_convnext_tiny_best_ema.safetensors",
"secondary_weights": "lodo_ddr_convnext_tiny_best.safetensors",
"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"
],
"held_out_fold": "ddr"
}
|