@@ -18,30 +18,17 @@ train_transform = [Rescale, RandomCrop, RandomFlip, NormalizeWithMeanStd]
1818valid_transform = [Rescale, CenterCrop, NormalizeWithMeanStd]
1919test_transform = [Rescale, CenterCrop, NormalizeWithMeanStd]
2020
21- Rescale_output_size = [256, 256]
22- Rescale_inverse = False
23-
21+ Rescale_output_size = [256, 256]
2422RandomCrop_output_size = [224, 224]
25- RandomCrop_foreground_focus = False
26- RandomCrop_foreground_ratio = None
27- RandomCrop_mask_label = None
28- RandomCrop_inverse = False
29-
3023CenterCrop_output_size = [224, 224]
31- CenterCrop_inverse = False
3224
3325RandomFlip_flip_depth = False
3426RandomFlip_flip_height = False
3527RandomFlip_flip_width = True
36- RandomFlip_inverse = False
3728
29+ NormalizeWithMeanStd_channels = [0, 1, 2]
3830NormalizeWithMeanStd_mean = [123.675, 116.28, 103.53]
3931NormalizeWithMeanStd_std = [58.395, 57.12, 57.375]
40- NormalizeWithMeanStd_channels = [0, 1, 2]
41- NormalizeWithMeanStd_mask = False
42- NormalizeWithMeanStd_random_fill = False
43- NormalizeWithMeanStd_inverse = False
44-
4532
4633[network]
4734# this section gives parameters for network
@@ -89,5 +76,4 @@ gpus = [0]
8976# checkpoint mode can be [0-latest, 1-best, 2-specified]
9077ckpt_mode = 1
9178output_csv = result/resnet18_ce1.csv
92- evaluation_mode = True
9379save_probability = True
0 commit comments