Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

High-Resolution Image Harmonization via Collaborative Dual Transformations

About

Given a composite image, image harmonization aims to adjust the foreground to make it compatible with the background. High-resolution image harmonization is in high demand, but still remains unexplored. Conventional image harmonization methods learn global RGB-to-RGB transformation which could effortlessly scale to high resolution, but ignore diverse local context. Recent deep learning methods learn the dense pixel-to-pixel transformation which could generate harmonious outputs, but are highly constrained in low resolution. In this work, we propose a high-resolution image harmonization network with Collaborative Dual Transformation (CDTNet) to combine pixel-to-pixel transformation and RGB-to-RGB transformation coherently in an end-to-end network. Our CDTNet consists of a low-resolution generator for pixel-to-pixel transformation, a color mapping module for RGB-to-RGB transformation, and a refinement module to take advantage of both. Extensive experiments on high-resolution benchmark dataset and our created high-resolution real composite images demonstrate that our CDTNet strikes a good balance between efficiency and effectiveness. Our used datasets can be found in https://github.com/bcmi/CDTNet-High-Resolution-Image-Harmonization.

Wenyan Cong, Xinhao Tao, Li Niu, Jing Liang, Xuesong Gao, Qihao Sun, Liqing Zhang• 2021

Related benchmarks

TaskDatasetResultRank
Image HarmonizationiHarmony4 HFlickr
MSE68.61
58
Image HarmonizationiHarmony4 Hday2night
MSE36.72
51
Image HarmonizationiHarmony4 HAdobe5k
MSE20.62
43
Image HarmonizationiHarmony4 HCOCO
MSE16.25
38
Image HarmonizationHAdobe5k iHarmony4 (test)
MSE21.24
37
Image HarmonizationiHarmony4
MSE23.75
27
Image HarmonizationHAdobe5K
PSNR38.77
11
Image HarmonizationiHarmony4 Hday2night 256x256 LR (test)
PSNR37.95
10
Image HarmonizationiHarmony4 HAdobe5K 256x256 LR (test)
PSNR38.24
10
Image HarmonizationiHarmony4 HFlickr 256x256 LR (test)
PSNR33.55
10
Showing 10 of 17 rows

Other info

Code

Follow for update