def rand_scale(image): scale = choice(np.arange(0.5, 0.9, 0.1)) return rescale(image, scale, preserve_range=True).astype(np.uint8)