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