| { | |
| "model_name": "efficientnet_b4_cbam", | |
| "num_classes": 5, | |
| "img_size": [224, 224], | |
| "mean": [0.485, 0.456, 0.406], | |
| "std": [0.229, 0.224, 0.225], | |
| "drop_rate": 0.3, | |
| "cbam_ratio": 16, | |
| "class_names": { | |
| "0": "No DR", | |
| "1": "Mild", | |
| "2": "Moderate", | |
| "3": "Severe", | |
| "4": "Proliferative DR" | |
| }, | |
| "default_weights": "efficientnet_b4_cbam_fold1.pth" | |
| } | |