baseline_Adam_finetune.json 1.28 KB
Newer Older
cc215's avatar
cc215 committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
{   "name": "Composite",
    "data": {
      "dataset_name":"UKBB" ,
      "readable_frames": ["ED", "ES"],
      "train_dir": "/vol/medic02/users/wbai/data/cardiac_atlas/UKBB_2964/sa/train",
      "validate_dir": "/vol/medic02/users/wbai/data/cardiac_atlas/UKBB_2964/sa/validation",
      "image_format_name": "sa_{frame}.nii.gz",
      "label_format_name": "label_sa_{frame}.nii.gz",
      "data_aug_policy" :"UKBB_advancedv2",
      "if_resample": true,
      "new_spacing": [1.25, 1.25, 10],
      "keep_z_spacing": true,
      "image_size":[224,224,1],
      "label_size":[224,224],
      "pad_size":  [192,192,1],
      "crop_size" :[192,192,1],
      "num_classes": 4,
      "use_cache": false,
      "myocardium_only": false,
      "ignore_black_slices": true

    },

 "segmentation_model": {
        "network_type": "UNet_64",
        "num_classes": 4,
        "resume_path":"./checkpoints/Unet_LVSA_trained_from_UKBB.pkl",
        "lr": 0.00001,
        "optimizer_name": "adam",
        "n_epochs": 1000,
        "max_iteration": 500000,
        "batch_size": 20,
        "use_gpu": true,
        "decoder_dropout": 0.2
    },

  "adversarial_augmentation":
  {
    "transformation_type":"composite"
  }
  ,
  "output":
  {
    "save_epoch_every_num_epochs":1,
     "save_dir":"./result/Composite"
  }
}