Ultra-High-Definition Image Restoration: New Benchmarks and A Dual Interaction Prior-Driven Solution

TL;DR

This paper introduces UHD snow and rain benchmarks and proposes a dual interaction prior-driven method for ultra-high-definition image restoration, achieving state-of-the-art results across multiple restoration tasks.

Contribution

The paper creates new UHD snow and rain benchmarks and develops a novel dual prior feature interaction method for UHD image restoration.

Findings

State-of-the-art performance on UHD image restoration tasks

Effective utilization of gradient and normal priors in model design

Introduction of new UHD benchmarks for snow and rain degradation

Abstract

Ultra-High-Definition (UHD) image restoration has acquired remarkable attention due to its practical demand. In this paper, we construct UHD snow and rain benchmarks, named UHD-Snow and UHD-Rain, to remedy the deficiency in this field. The UHD-Snow/UHD-Rain is established by simulating the physics process of rain/snow into consideration and each benchmark contains 3200 degraded/clear image pairs of 4K resolution. Furthermore, we propose an effective UHD image restoration solution by considering gradient and normal priors in model design thanks to these priors' spatial and detail contributions. Specifically, our method contains two branches: (a) feature fusion and reconstruction branch in high-resolution space and (b) prior feature interaction branch in low-resolution space. The former learns high-resolution features and fuses prior-guided low-resolution features to reconstruct clear images, while the latter utilizes normal and gradient priors to mine useful spatial features and detail features to guide high-resolution recovery better. To better utilize these priors, we introduce single prior feature interaction and dual prior feature interaction, where the former respectively fuses normal and gradient priors with high-resolution features to enhance prior ones, while the latter calculates the similarity between enhanced prior ones and further exploits dual guided filtering to boost the feature interaction of dual priors. We conduct experiments on both new and existing public datasets and demonstrate the state-of-the-art performance of our method on UHD image low-light enhancement, dehazing, deblurring, desonwing, and deraining. The source codes and benchmarks are available at \url{https://github.com/wlydlut/UHDDIP}.