Search papers, labs, and topics across Lattice.
This paper introduces BoxCtrl, a novel 3D-aware visual prompting framework designed to enhance geometric image editing tasks such as translation, scaling, and rotation in 3D space. By utilizing RGB 3D bounding boxes projected onto 2D images, BoxCtrl effectively decouples geometric control from appearance control, allowing for precise correspondences in the latent space. The two-stage training approach, which combines Supervised Fine-Tuning with Reinforcement Learning, results in significant improvements in geometric accuracy and visual fidelity, achieving state-of-the-art performance in various editing tasks.
BoxCtrl's innovative use of RGB 3D bounding boxes enables unprecedented precision in geometric image editing, outperforming traditional methods.
As instruction-based editing models and multimodal large language models advance, diverse image editing tasks have become feasible. However, achieving precise and consistent geometric image editing, such as translating, scaling, and rotating in 3D space, remains a major challenge. In this work, we introduce BoxCtrl, a 3D-aware visual prompting framework. Unlike text-only or coarse 2D-guided approaches, our method introduces informative RGB 3D bounding boxes projected onto 2D images as visual prompts. The three orthogonal faces of each box are painted with distinct RGB colors, simultaneously encoding position, size, and orientation to provide a compact, intuitive in-context visual example. The key to BoxCtrl's success lies in these well-designed bounding boxes, which decouple geometric control from appearance control. This enables the model to learn consistent correspondences between faces of the same color in the latent space, leading to a precise understanding of geometric intentions and accurate editing results. We introduce a two-stage training paradigm: Supervised Fine-Tuning (SFT) followed by Reinforcement Learning (RL). To address paired data scarcity, we construct a large-scale synthetic dataset for SFT, equipping the model with fundamental editing capabilities. To bridge the synthetic-to-real domain gap, we incorporate an online RL stage leveraging unpaired real-world data. Guided by a reward function evaluating geometric accuracy and visual fidelity, our SFT-RL strategy significantly enhances geometric precision while maintaining photorealistic quality. Extensive experiments demonstrate that BoxCtrl achieves state-of-the-art performance across translation, rotation, scaling, and composite editing tasks.
