def TF_shift_hue(x, shift=0.0): assert len(x.shape) == 3 h, w, nc = x.shape hsv = rgb2hsv(x) hsv[:,:,0] += shift return hsv2rgb(hsv)