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)
微信
公众号
扫码关注公众号