Here is a problem setting:
Function that converts from rgb to hsv:
def rbg2hsv(img_rgb): batch_size, channel, height, width = img_rgb.size() r, g,