def load_image(f, mode='RGB'):
    with Image.open(f) as image:
        return np.array(image.convert(mode))
def load_image(f, mode='RGB'):
    with Image.open(f) as image:
        return np.array(image.convert(mode))