EASI-Tex: Edge-Aware Mesh Texturing from Single Image

Sai Raj Kishore Perla · Yizhi Wang · Ali Mahdavi-Amiri · Hao (Richard) Zhang
ACM Transactions on Graphics (Proceedings of SIGGRAPH), 2024

arXiv | Project Page

$Logo$

We present a novel approach for single-image mesh texturing, which employs a diffusion model with judicious conditioning to seamlessly transfer an object's texture from a single RGB image to a given 3D mesh object. We do not assume that the two objects belong to the same category, and even if they do, there can be significant discrepancies in their geometry and part proportions. Our method aims to rectify the discrepancies by conditioning a pre-trained Stable Diffusion generator with edges describing the mesh through ControlNet, and features extracted from the input image using IP-Adapter to generate textures that respect the underlying geometry of the mesh and the input texture without any optimization or training. We also introduce Image Inversion, a novel technique to quickly personalize the diffusion model for a single concept using a single image, for cases where the pre-trained IP-Adapter falls short in capturing all the details from the input image faithfully. Experimental results demonstrate the efficiency and effectiveness of our edge-aware single-image mesh texturing approach, coined EASI-Tex, in preserving the details of the input texture on diverse 3D objects, while respecting their geometry.

Setup

The code is tested on Ubuntu 20.04 LTS with PyTorch 2.0.1 and CUDA 11.8. Please follow the steps below to setup the environment.

# create and activate the conda environment
conda create -n easitex python=3.10
conda activate easitex

# install PyTorch 2.0.1
conda install pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.8 -c pytorch -c nvidia

Then, install PyTorch3D:

# install runtime dependencies for PyTorch3D
conda install -c fvcore -c iopath -c conda-forge fvcore iopath
conda install -c bottler nvidiacub

# install PyTorch3D
conda install pytorch3d -c pytorch3d

Install xformers to accelerate transformers:

# please don't use pip to install it, as it only supports PyTorch>=2.0
conda install xformers -c xformers

Install the necessary packages listed out in requirements.txt:

pip install -r requirements.txt

Download the following files:

control_sd15_canny.pth from ControlNet, and put it under models/ControlNet/models/.
control_v11p_sd15_canny.pth from ControlNet v1.1, and put it under models/ControlNet/models/.
image_encoder/ from IP-Adapter, and put it under ip_adapter/.
ip-adapter-plus_sd15.bin from IP-Adapter, and put it under ip_adapter/.

Usage

Try the demo / Sanity check

To make sure everything is set up and configured correctly, you can run the following script to transfer texture from a round bird to a cow.

./bash/run.sh

Once the synthesis completes, you should be able to see the generated assets under outputs/canny/cow-round_bird/42-ip1.0-cn1.0-dist0.8-gs10.0-p36-h20-us0.4-vt0.1. The final textured mesh 42-ip1.0-cn1.0-dist0.8-gs10.0-p36-h20-us0.4-vt0.1/update/mesh/19.obj should look something similar to:

Try on your own data

Mesh pre-processing: For best results, make sure the input mesh satisfies the following criteria:

It is triangulated.
+Y is up.
+Z is front.

One can use the scripts provided in scripts/, or use 3D softwares such as Meshlab or Blender to pre-process the input mesh. A mesh ready for next steps should look like:

Image pre-processing: We need an RGB image and a corresponding mask for the region of interest in the image to transfer texture from. The image should be named as <img_name>.jpg and the mask should be named as <img_name>_mask.jpg, and both the images are expected to be in the same directory. See data/texture_images/round_bird for reference.

Texture transfer: Use the bash script bash/run.sh to call scripts/generate_texture.py and transfer textures from an image to a given 3D mesh. Set --tex_resolution to "3k" to generate high-resolution (3k) textures, however it requires significantly more GPU memory than generating "1k" textures.

Please check scripts/generate_texture.py, or simply run python scripts/generate_texture.py -h, to see all the available arguments.

Acknowledgement

Our code builds upon Text2Tex and Diffusers_IPAdapter, we thank their authors for providing such amazing codebases. Additionally, the image of the round bird and the cow mesh were sourced from here and here respectively.

Citation

If you found our work helpful, please consider citing:

@article{perla2024easitex,
    title={EASI-Tex: Edge-Aware Mesh Texturing from Single Image},
    author = {Perla, Sai Raj Kishore and Wang, Yizhi and Mahdavi-Amiri, Ali and Zhang, Hao},
    journal = {ACM Transactions on Graphics (Proceedings of SIGGRAPH)},
    publisher = {ACM New York, NY, USA},
    year = {2024},
    volume = {43},
    number = {4},
    articleno = {40},
    doi = {10.1145/3658222},
    url = {https://github.com/sairajk/easi-tex},
}