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

Before Change

    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:
            image_mask = np.asarray(mask, dtype=np.bool)

After Change

            image_5d = np.asarray(image, dtype=np.float32)

        image_mask = None
        if isinstance(mask, dict):
            image_mask = mask.get(self.field, None)
        elif mask is not None:
            image_mask = mask
        elif self.binary_masking_func is not None:
            image_mask = self.binary_masking_func(image_5d)
        else:
            // no access to mask, default to all image
            image_mask = np.ones_like(image_5d, dtype=np.bool)

        normalised = self._normalise_5d(image_5d, image_mask)

        if isinstance(image, dict):
            image[self.field] = normalised

Instances