label normalisation conditions
@@ -113,7 +113,8 @@ class SegmentationApplication(BaseApplication):
normalisation_layers.append(histogram_normaliser)
if self.net_param.whitening:
normalisation_layers.append(mean_var_normaliser)
- if task_param.label_normalisation:
+ if task_param.label_normalisation and \
+ (self.is_training or not task_param.output_prob):
normalisation_layers.append(label_normaliser)
augmentation_layers = []
