ebbc9fc0fc52a650ebd5bbbd954733bada35672e,niftynet/layer/histogram_normalisation.py,HistogramNormalisationLayer,layer_op,#HistogramNormalisationLayer#Any#Any#,58

Before Change

        self.modalities = modalities

    def layer_op(self, image_5d, mask=None):
        assert image_5d.ndim == 5
        image_5d = np.asarray(image_5d, dtype=float)

        image_mask = None
        if mask is not None:

After Change

        normalised = self._normalise_5d(image_5d, image_mask)

        if isinstance(image, dict):
            image[self.field] = normalised
            if isinstance(mask, dict):
                mask[self.field] = image_mask
            else:
                mask = {self.field: image_mask}
            return image, mask
        else:
            return normalised, image_mask

    def __check_modalities_to_train(self):
        modalities_to_train = [mod for mod in self.modalities
                               if not mod in self.mapping]
        return set(modalities_to_train)

Instances