config.json

{
  "model_type": "resnet3d",
  "task": "normal_vs_rd",
  "pipeline_tag": "image-classification",
  "num_labels": 2,
  "id2label": {
    "0": "Normal",
    "1": "Retinal Detachment"
  },
  "label2id": {
    "Normal": 0,
    "Retinal Detachment": 1
  },
  "input_modality": "3D ocular ultrasound video",
  "input_shape": [
    1,
    96,
    128,
    128
  ],
  "input_format": "video (mp4), normalized by /255.0",
  "architecture": {
    "in_channels": 1,
    "num_classes": 1,
    "block": "basic",
    "layers": [
      4,
      4,
      4,
      4
    ],
    "block_inplanes": [
      64,
      128,
      256,
      512
    ]
  },
  "pooling": "avg"
}