cooperative_training.json

{
  "name": "cooperative training",
  "data": {
    "dataset_name": "ACDC",
    "root_dir": "/vol/biomedic3/cc215/data/MICCAI2021_multi_domain_robustness_datasets/ACDC",
    "frame": [
      "ES",
      "ED"
    ],
    "image_size": [
      224,
      224,
      1
    ],
    "label_size": [
      224,
      224
    ],
    "pad_size": [
      224,
      224,
      1
    ],
    "crop_size": [
      192,
      192,
      1
    ],
    "data_aug_policy": "ACDC_affine_elastic_intensity",
    "image_format_name": "{p_id}/{frame}_img.nii.gz",
    "label_format_name": "{p_id}/{frame}_seg.nii.gz",
    "num_classes": 4,
    "use_cache": true,
    "keep_orig_image_label_pair_for_training": true,
    "myocardium_only": false,
    "right_ventricle_only": false
  },
  "segmentation_model": {
    "network_type": "FCN_16_standard",
    "num_classes": 4
  },
  "learning": {
    "latent_DA": true,
    "separate_training": false,
    "lr": 0.0001,
    "n_epochs": 600,
    "max_iteration": 50000,
    "batch_size": 20,
    "use_gpu": true
  },
  "latent_DA": {
    "mask_scope": [
      "image code",
      "shape code"
    ],
    "image code": {
      "loss_name": "mse",
      "mask_type": "random",
      "max_threshold": 0.5,
      "random_threshold": true,
      "if_soft": true
    },
    "shape code": {
      "loss_name": "ce",
      "mask_type": "random",
      "max_threshold": 0.5,
      "random_threshold": true,
      "if_soft": true
    }
  },
  "output": {
    "save_epoch_every_num_epochs": 100
  }
}