config.json

{
  "architectures": ["ResNet18"],
  "model_type": "resnet",
  "num_classes": 5,
  "hidden_sizes": [256, 128, 64],
  "image_size": 512,
  "num_channels": 3,
  "id2label": {
    "0": "Normal",
    "1": "Mild",
    "2": "Moderate",
    "3": "Severe",
    "4": "Proliferative"
  },
  "label2id": {
    "Normal": 0,
    "Mild": 1,
    "Moderate": 2,
    "Severe": 3,
    "Proliferative": 4
  }
}