config.json

{
  "model_type": "organscan",
  "architectures": [
    "MultiHeadAnatomyNet"
  ],
  "backbone": "mobilenetv4_conv_small.e2400_r224_in1k",
  "image_size": 224,
  "heads": {
    "modality": 5,
    "region": 8,
    "organ": 30
  },
  "taxonomy_version": "1.0.0",
  "outputs": [
    "modality_logits",
    "region_logits",
    "organ_logits"
  ],
  "input": {
    "name": "pixel_values",
    "shape": [
      "batch",
      3,
      224,
      224
    ]
  }
}