Diff-FCCF/config.json at main · DanceSkyCode/Diff-FCCF · GitHub

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
48
49
50
51
52
53
54
55
56
{
  "name": "Diff-FCCF",
  "hyperParameter": {
      "epochs": 10000,
      "start_epoch": 0,
      "loss_step": 67,
      "save_model_epoch_step": 100
  },
  "dataset": {
      "train": { "path":"Dataset/Medical/Train",
                "phase": "train",
                "batch_size": 4,
                "shuffle": true,
                "drop_last": true,
                "use_dataTransform": true,
                "resize": true,
                "imgSize": 256
      },
      "valid": { "dataset_name": "VDMUFusion",
                "path": "/Havard-Medical-Image-Fusion-Datasets-main/MyDatasets/PET-MRI/test1",
                "phase": "valid",
                "batch_size": 1,
                "shuffle": false,
                "drop_last": false,
                "use_dataTransform": true,
                "resize": false,
                "imgSize": -1,
                "generat_imgs_num": 1
      }
  },
  "diffusion_model": {
       "T": 2000,
       "beta_schedule_type": "cosine",
       "add_noise": false,
       "loss_scale": 128
  },
  "Condition_Noise_Predictor": {
      "use_preTrain_model": true,
      "preTrain_Model_path": "/weight/20250424_005754/epoch_1100.pt",
      "concat_type": "ABX",
      "UNet": { "in_channels": 4,
                "out_channels": 1,
                "model_channels": 64,
                "num_res_blocks": 2,
                "dropout": 0.1,
                "time_embed_dim_mult":4,
                "down_sample_mult": [1, 2, 4, 8]
      }
  },
  "optimizer": {
      "init_lr": 2e-4,
      "use_lr_scheduler": true,
      "StepLR_size": 1000,
      "StepLR_gamma": 0.99
  }
}