OmniPaint: Mastering Object-Oriented Editing via Disentangled Insertion-Removal Inpainting

Yongsheng Yu1*  Ziyun Zeng1*  Haitian Zheng2  Jiebo Luo1 
1University of Rochester   2Adobe Research  

Abstract

Diffusion-based generative models have revolutionized object-oriented image editing, yet their deployment in realistic object removal and insertion remains hampered by challenges such as the intricate interplay of physical effects and insufficient paired training data.

In this work, we introduce OmniPaint, a unified framework that re-conceptualizes object removal and insertion as interdependent processes rather than isolated tasks. Leveraging a pre-trained diffusion prior along with a progressive training pipeline comprising initial paired sample optimization and subsequent large-scale unpaired refinement via CycleFlow, OmniPaint achieves precise foreground elimination and seamless object insertion while faithfully preserving scene geometry and intrinsic properties.

Furthermore, our novel CFD metric offers a robust, reference-free evaluation of context consistency and object hallucination, establishing a new benchmark for high-fidelity image editing.


Realistic Object Removal

Input Image 7 Output Image 7
Input Image 11 Output Image 11
Input Image 1 Output Image 1
Input Image 8 Output Image 8
Input Image 9 Output Image 9
Input Image 20 Output Image 20
Input Image 13 Output Image 13
Input Image 15 Output Image 15
Input Image 17 Output Image 17
Input Image 6 Output Image 6
Input Image 21 Output Image 21

Generative Object Insertion

Subject Image 25
Input Image 25 Output Image 25
Input Image 28 Output Image 28
Subject Image 28
Subject Image 14-1
Subject Image 14-2
Input Image 14 Output Image 14-1
Input Image 22 Output Image 22-2
Subject Image 22-2
Subject Image 22-1

BibTeX

@article{yu2025omnipaint,
  title={OmniPaint: Mastering Object-Oriented Editing via Disentangled Insertion-Removal Inpainting},
  author={Yu, Yongsheng and Zeng, Ziyun and Zheng, Haitian and Luo, Jiebo},
  journal={arXiv preprint arXiv:2503.08677},
  year={2025}
}