def save_image()

in test.py [0:0]


def save_image(path, image):
    imageio.imwrite(path, image_to_np(image))