2511.12785

Total: 1

#1 Lightweight Optimal-Transport Harmonization on Edge Devices [PDF] [Copy] [Kimi] [REL]

Authors: Maria Larchenko, Dmitry Guskov, Alexander Lobashev, Georgy Derevyanko

Color harmonization adjusts the colors of an inserted object so that it perceptually matches the surrounding image, resulting in a seamless composite. The harmonization problem naturally arises in augmented reality (AR), yet harmonization algorithms are not currently integrated into AR pipelines because real-time solutions are scarce. In this work, we address color harmonization for AR by proposing a lightweight approach that supports on-device inference. For this, we leverage classical optimal transport theory by training a compact encoder to predict the Monge-Kantorovich transport map. We benchmark our MKL-Harmonizer algorithm against state-of-the-art methods and demonstrate that for real composite AR images our method achieves the best aggregated score. We release our dedicated AR dataset of composite images with pixel-accurate masks and data-gathering toolkit to support further data acquisition by researchers.

Subjects: Computer Vision and Pattern Recognition , Artificial Intelligence , Graphics

Publish: 2025-11-16 21:29:46 UTC