9bf2004b29aa393400bd904bdf788390f5c0404a,niftynet/layer/histogram_normalisation.py,HistogramNormalisationLayer,__normalise_5d,#HistogramNormalisationLayer#Any#Any#,98

Before Change

        assert data_array.shape[3] <= len(self.modalities)

        if self.mapping is {}:
            raise RuntimeError("calling normaliser with empty mapping,"
                               "probably {} is not loaded".format(
                self.hist_model_file))
        for mod in self.modalities:
            for t in range(0, data_array.shape[4]):
                mod_id = self.modalities[mod]
                if not np.any(data_array[..., mod_id, t]):
                    continue  // missing modality
                data_array[..., mod_id, t] = self.__normalise_3d(
                    data_array[..., mod_id, t],

After Change

                "probably {} is not loaded".format(self.hist_model_file))
            raise RuntimeError
        for t in range(0, data_array.shape[3]):
            for mod_id, mod_name in enumerate(self.modalities):
                if not np.any(data_array[..., t, mod_id]):
                    continue  // missing modality
                data_array[..., t, mod_id] = self.__normalise_3d(
                    data_array[..., t, mod_id],
                    mask_array[..., t, mod_id],
                    self.mapping[mod_name])
        return data_array

    def __normalise_3d(self, img_data, mask, mapping):
        assert img_data.ndim == 3

Instances