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