SHAP-EDITOR: Instruction-guided Latent 3D Editing in Seconds (CVPR 2024)

Minghao Chen, Junyu Xie, Iro Laina, Andrea Vedaldi

Paper | Webpage | Demo

We present a method, named Shap-Editor, aiming at fast 3D editing. We propose to learn a universal editing function that can be applied to different objects within one second.

News:

• Jun 18, 2024 🔥 🔥 : The inference code is available. Try to modify your own 3D object in seconds!
• Feb 26, 2024 🔥: Shap-Editor is accepted in CVPR 2024!!!
• Dec 15, 2023: Demo is available here on Hugging Face.

Environment Setup

To set up the environment you can easily run the following command:

conda create -n Shap-Editor python=3.9
conda activate Shap-Editor
pip install -r requirements.txt

Pre-trained Models

We provide some pre-trained editing models for playing around.

prompt	Link	Model Type
"Add a santa hat to it"	Link	Single
"Make it look like made of gold"	Link	Single
"Make the color of it look like rainbow"	Link	Single
"Make it in cyberpunk style"	Link	Single
"Make it wooden"	Link	Single
"Make it look like make of lego"	Link	Single

Demo

We provide demo in following ways:

1. You can try our Gradio demo on Hugging Face. (Thank Hugging Face for providing free GPU support!)
2. You can also run the demo locally using [python app.py].

Inference

Step 1: Get the latent

To edit a 3D assets, we need its corresponding latent. In this work, we consider two main sources of latents:

1. Latents generate by Shap-E diffusion models. You can generate your own 3D assets with latent by use:

python generate_3d.py --prompt 'a shark' -output_dir PATH_TO_OUTPUT_DIR --export_mesh

2. Encoded Latents from real 3D assets. You can encode your own 3D assets into latents following Shap-E! As Shap-E requires blender to render multi-view images for 3D encoding, you need to first install Blender version >= 3.3.1, and set the environment variable BLENDER_PATH to the path of the Blender executable. For more details, please refer to the original repo. After that you can simply run to encode the 3D asset:

python encode_3d.py --latent_path PATH_TO_YOUR_OBJ --output_dir PATH_TO_OUTPUT_DIR

Step 2: Edit the latent with pre-trained Shap-Editor

After get the latent of the 3D asset, you can edit the latent within 1 sec using the pre-trained editing model or editing model you trained yourself.

Citation

If this repo is helpful for you, please consider to cite it. Thank you! :)

@inproceedings{chen2024shap,
  title={SHAP-EDITOR: Instruction-guided Latent 3D Editing in Seconds},
  author={Chen, Minghao and Xie, Junyu and Laina, Iro and Vedaldi, Andrea},
  booktitle={Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition},
  pages={26456--26466},
  year={2024}
}

To Do List

• Demo
• Inference code
• Training code

Acknowledgements

This research is supported by ERC-CoG UNION 101001212. Iro Laina is also partially supported by the VisualAI EPSRC grant (EP/T028572/1).

The code is largely based on Shap-E. It is also inspired by worderful projects:

• InstructPix2Pix
• Diffusers