config.json

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