main.py 文件源码

def load_image(path):
    img = skimage.io.imread(path)
    yuv = cv2.cvtColor(np.float32(img), cv2.COLOR_RGB2YUV)
    img = img - vgg19.VGG_MEAN
    img = img[:,:,(2,1,0)]  # rgb to bgr
    return img[np.newaxis, :, :, :], yuv