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

Neural Preset for Color Style Transfer

About

In this paper, we present a Neural Preset technique to address the limitations of existing color style transfer methods, including visual artifacts, vast memory requirement, and slow style switching speed. Our method is based on two core designs. First, we propose Deterministic Neural Color Mapping (DNCM) to consistently operate on each pixel via an image-adaptive color mapping matrix, avoiding artifacts and supporting high-resolution inputs with a small memory footprint. Second, we develop a two-stage pipeline by dividing the task into color normalization and stylization, which allows efficient style switching by extracting color styles as presets and reusing them on normalized input images. Due to the unavailability of pairwise datasets, we describe how to train Neural Preset via a self-supervised strategy. Various advantages of Neural Preset over existing methods are demonstrated through comprehensive evaluations. Notably, Neural Preset enables stable 4K color style transfer in real-time without artifacts. Besides, we show that our trained model can naturally support multiple applications without fine-tuning, including low-light image enhancement, underwater image correction, image dehazing, and image harmonization. Project page with demos: https://zhkkke.github.io/NeuralPreset .

Zhanghan Ke, Yuhao Liu, Lei Zhu, Nanxuan Zhao, Rynson W.H. Lau• 2023

Related benchmarks

TaskDatasetResultRank
Image HarmonizationiHarmony4
MSE24.31
27
Image Color Style TransferLuan et al. (test)
CPU Inference Time (s)0.215
15
Style TransferTST2K
CP Count85.79
15
Style TransferPST50
CP Count79.19
15
Tone Style TransferPST50
CP0.6914
15
Tone Style TransferTST2K
PSNR23.06
14
Low-light Image EnhancementLOL and SICE Averaged v1, v2 (test)
Content Similarity0.988
9
Tone manipulationMIT Adobe FiveK
Content Sim.99.2
9
Photorealistic Preset TransferRealistic Dataset (test)
GPT-4o Score2.7667
9
Photorealistic Preset TransferSynthetic dataset
PSNR22.16
9
Showing 10 of 22 rows

Other info

Code

Follow for update