pradhaansbhat/Thinking-In-Boxes

[NeurIPS-2026] Thinking in Boxes: 3D Editing in Real Images Made Easy

Pradhaan S Bhat^1∗ · Naveen Chandra R^1∗ · Rishubh Parihar¹ · Vaibhav Vavilala² · R. Venkatesh Babu¹ · D.A. Forsyth · Anand Bhattad⁴

¹ Indian Institute of Science ² Apple ³ UIUC ⁴ Johns Hopkins University

^∗ Equal Contribution

This is the trained LoRA used in the paper Thinking In Boxes: 3D Editing in Real Images Made Easy.

Thinking-In-Boxes is an Image-to-Image Generative LoRA for black-forest-labs/FLUX.1-Kontext-dev trained for the task of Geometric Image Editing.

This LoRA is trained on the Thinking-In-Boxes dataset available here. Details of training are available in the supplementary section of the paper.

Usage with Diffusers 🧨

Please refer to the GitHub Repository on setup, inference and training.

Citation

If you find our work useful, please consider citing:

@misc{bhat2026thinkingboxes3dediting,
  title         = {Thinking in Boxes: 3D Editing in Real Images Made Easy},
  author        = {Pradhaan S Bhat and Naveen Chandra R and Rishubh Parihar and Vaibhav Vavilala and R. Venkatesh Babu and D. A. Forsyth and Anand Bhattad},
  year          = {2026},
  eprint        = {2606.20556},
  archivePrefix = {arXiv},
  primaryClass  = {cs.CV},
  url           = {https://arxiv.org/abs/2606.20556}
}

---