|
{ |
|
"batch_size": 32, |
|
"sliding_window_mode": "constant", |
|
"models": { |
|
"lr": "1e-5", |
|
"wd": 0, |
|
"dropout": 0.0, |
|
"model": "unet", |
|
"unet": { |
|
"channels": [ |
|
32, |
|
64, |
|
128, |
|
256 |
|
], |
|
"strides": [ |
|
2, |
|
2, |
|
2 |
|
], |
|
"activation": "LEAKYRELU", |
|
"normalization": "BATCH" |
|
} |
|
}, |
|
"augmentation": { |
|
"patch_size": [ |
|
64, |
|
64, |
|
64 |
|
], |
|
"use_augmentation": true, |
|
"p_gaussian_noise": 0.5, |
|
"p_gaussian_smooth": 0.5, |
|
"p_intensity_scale": 0, |
|
"p_intensity_shift": 0, |
|
"p_adjust_contrast": 0, |
|
"p_histogram_shift": 0.5, |
|
"p_rotate": 0.5, |
|
"p_zoom": 0.5, |
|
"p_shear": 0, |
|
"p_flip": [ |
|
0, |
|
0.5, |
|
0.5 |
|
], |
|
"gaussian_noise_mean": 0.0, |
|
"gaussian_noise_std": 0.01, |
|
"gaussian_smooth_sigma": [ |
|
0.0, |
|
0.5 |
|
], |
|
"intensity_scale_factors": [ |
|
-0.5, |
|
1.2 |
|
], |
|
"intensity_shift_offsets": [ |
|
0.05, |
|
0.2 |
|
], |
|
"adjust_contrast_gamma": [ |
|
0.7, |
|
1.3 |
|
], |
|
"control_points": [ |
|
5, |
|
10 |
|
], |
|
"rotation": [ |
|
-90, |
|
90 |
|
], |
|
"min_zoom": 1.0, |
|
"max_zoom": 1.2, |
|
"shear_range": [ |
|
0.1, |
|
0.1, |
|
0.1 |
|
] |
|
} |
|
} |