@@ -10,7 +10,6 @@ test_csv = ../JSRT/config/jsrt_test.csv
1010
1111train_batch_size = 4
1212
13- load_pixelwise_weight = False
1413# modality number
1514modal_num = 1
1615
@@ -20,24 +19,13 @@ valid_transform = [NormalizeWithMeanStd, LabelConvert, LabelToProbability]
2019test_transform = [NormalizeWithMeanStd]
2120
2221NormalizeWithMeanStd_channels = [0]
23- NormalizeWithMeanStd_mean = None
24- NormalizeWithMeanStd_std = None
25- NormalizeWithMeanStd_mask = False
26- NormalizeWithMeanStd_random_fill = False
27- NormalizeWithMeanStd_inverse = False
2822
2923LabelConvert_source_list = [0, 255]
3024LabelConvert_target_list = [0, 1]
31- LabelConvert_inverse = False
3225
3326RandomCrop_output_size = [240, 240]
34- RandomCrop_foreground_focus = False
35- RandomCrop_foreground_ratio = None
36- RandomCrop_mask_label = None
37- RandomCrop_inverse = False
3827
3928LabelToProbability_class_num = 2
40- LabelToProbability_inverse = False
4129
4230[network]
4331# this section gives parameters for network
@@ -50,7 +38,7 @@ net_type = MyUNet2D
5038class_num = 2
5139in_chns = 1
5240feature_chns = [4, 16, 24, 32, 48]
53- dropout = [0.0, 0.0, 0.3, 0.3, 0. 4, 0.5]
41+ dropout = [0.0, 0.0, 0.3, 0.4, 0.5]
5442bilinear = True
5543
5644[training]
@@ -64,9 +52,6 @@ MyFocalDiceLoss_Enable_Class_Weight = True
6452MyFocalDiceLoss_beta = 1.5
6553class_weight = [0.2, 1.0]
6654
67- CrossEntropyLoss_Enable_Pixel_Weight = False
68- CrossEntropyLoss_Enable_Class_Weight = False
69-
7055# for optimizers
7156optimizer = Adam
7257learning_rate = 1e-3
@@ -94,12 +79,6 @@ gpus = [0]
9479# checkpoint mode can be [0-latest, 1-best, 2-specified]
9580ckpt_mode = 0
9681output_dir = result
97- evaluation_mode = True
98- test_time_dropout = False
99-
100- # use test time augmentation
101- tta_mode = 0
102- infer_sliding_window = False
10382
10483# convert the label of prediction output
10584label_source = [0, 1]
0 commit comments