High-Resolution Photorealistic Image Translation in Real-Time: A Laplacian Pyramid Translation Network

CVPR 2021  ยท  Jie Liang, Hui Zeng, Lei Zhang ยท

Existing image-to-image translation (I2IT) methods are either constrained to low-resolution images or long inference time due to their heavy computational burden on the convolution of high-resolution feature maps. In this paper, we focus on speeding-up the high-resolution photorealistic I2IT tasks based on closed-form Laplacian pyramid decomposition and reconstruction. Specifically, we reveal that the attribute transformations, such as illumination and color manipulation, relate more to the low-frequency component, while the content details can be adaptively refined on high-frequency components. We consequently propose a Laplacian Pyramid Translation Network (LPTN) to simultaneously perform these two tasks, where we design a lightweight network for translating the low-frequency component with reduced resolution and a progressive masking strategy to efficiently refine the high-frequency ones. Our model avoids most of the heavy computation consumed by processing high-resolution feature maps and faithfully preserves the image details. Extensive experimental results on various tasks demonstrate that the proposed method can translate 4K images in real-time using one normal GPU while achieving comparable transformation performance against existing methods. Datasets and codes are available: https://github.com/csjliang/LPTN.

PDF Abstract CVPR 2021 PDF CVPR 2021 Abstract

Datasets


Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Photo Retouching MIT-Adobe 5k LPTN (L=3) PSNR 22.02 # 5
SSIM 0.879 # 5
Photo Retouching MIT-Adobe 5k (1080p) LPTN (L=3) PSNR 22.09 # 1
SSIM 0.883 # 2
Photo Retouching MIT-Adobe 5k (1080p) DPE PSNR 21.94 # 2
SSIM 0.885 # 1
Photo Retouching MIT-Adobe 5k (480p) LPTN (L=3) PSNR 22.12 # 1
SSIM 0.878 # 1
Photo Retouching MIT-Adobe 5k (480p) DPE PSNR 21.99 # 2
SSIM 0.875 # 2

Methods