def postprocess_image()

in data_preparation/image_utils.py [0:0]


def postprocess_image(image: tf.Tensor) -> Image.Image:
    output = (image[0].numpy() + 1.0) * 127.5
    output = np.clip(output, 0, 255).astype(np.uint8)
    output = cv2.cvtColor(output, cv2.COLOR_BGR2RGB)
    output_image = Image.fromarray(output)
    return output_image