config.json

{
  "task": "multilabel-classification",
  "num_classes": 6,
  "classes": [
    "epidural",
    "intraparenchymal",
    "intraventricular",
    "subarachnoid",
    "subdural",
    "any"
  ],
  "img_size": 380,
  "models": [
    {
      "name": "tf_efficientnet_b4.ns_jft_in1k",
      "file": "tf_efficientnet_b4_ns_jft_in1k_fold0.pt"
    },
    {
      "name": "convnext_small.fb_in22k_ft_in1k",
      "file": "convnext_small_fb_in22k_ft_in1k_fold0.pt"
    },
    {
      "name": "resnet18",
      "file": "ich_resnet18.pt"
    }
  ]
}