def apply_p2_98(data): """ Image normalization """ p2 = np.percentile(data, 2) p98 = np.percentile(data, 98) data = exposure.rescale_intensity(data, in_range=(p2, p98)) return data