Input image prediction