HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

High-Resolution Image Harmonization via Collaborative Dual Transformations

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

High-Resolution Image Harmonization via Collaborative Dual Transformations

Abstract

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.

Code Repositories

bcmi/CDTNet-High-Resolution-Image-Harmonization
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
image-harmonization-on-hadobe5k-1024-timesCDTNet
MSE: 21.24
PSNR: 38.77
SSIM: 0.9868
fMSE: 152.13
image-harmonization-on-iharmony4CDTNet
MSE: 23.75
PSNR: 38.23
fMSE: -

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
High-Resolution Image Harmonization via Collaborative Dual Transformations | Papers | HyperAI