Awesome
Text2Mesh [Project Page]
Text2Mesh is a method for text-driven stylization of a 3D mesh, as described in "Text2Mesh: Text-Driven Neural Stylization for Meshes" CVPR 2022.
Getting Started
Installation
Note: The below installation will fail if run on something other than a CUDA GPU machine.
conda env create --file text2mesh.yml
conda activate text2mesh
If you experience an error installing kaolin saying something like nvcc not found
, you may need to set your CUDA_HOME
environment variable to the 11.3 folder i.e. export CUDA_HOME=/usr/local/cuda-11.3
, then rerunning the installation.
System Requirements
- Python 3.7
- CUDA 11
- GPU w/ minimum 8 GB ram
Run examples
Call the below shell scripts to generate example styles.
# cobblestone alien
./demo/run_alien_cobble.sh
# shoe made of cactus
./demo/run_shoe.sh
# lamp made of brick
./demo/run_lamp.sh
# ...
The outputs will be saved to results/demo
, with the stylized .obj files, colored and uncolored render views, and screenshots during training.
Outputs
<p float="center"> <img alt="alien" height="135" src="images/alien.png" width="240"/> <img alt="alien geometry" height="135" src="images/alien_cobble_init.png" width="240"/> <img alt="alien style" height="135" src="images/alien_cobble_final.png" width="240"/> </p> <p float="center"> <img alt="alien" height="135" src="images/alien.png" width="240"/> <img alt="alien geometry" height="135" src="images/alien_wood_init.png" width="240"/> <img alt="alien style" height="135" src="images/alien_wood_final.png" width="240"/> </p> <p float="center"> <img alt="candle" height="135" src="images/candle.png" width="240"/> <img alt="candle geometry" height="135" src="images/candle_init.png" width="240"/> <img alt="candle style" height="135" src="images/candle_final.png" width="240"/> </p> <p float="center"> <img alt="person" height="135" src="images/person.png" width="240"/> <img alt="ninja geometry" height="135" src="images/ninja_init.png" width="240"/> <img alt="ninja style" height="135" src="images/ninja_final.png" width="240"/> </p> <p float="center"> <img alt="shoe" height="135" src="images/shoe.png" width="240"/> <img alt="shoe geometry" height="135" src="images/shoe_init.png" width="240"/> <img alt="shoe style" height="135" src="images/shoe_final.png" width="240"/> </p> <p float="center"> <img alt="vase" height="135" src="images/vase.png" width="240"/> <img alt="vase geometry" height="135" src="images/vase_init.png" width="240"/> <img alt="vase style" height="135" src="images/vase_final.png" width="240"/> </p> <p float="center"> <img alt="lamp" height="135" src="images/lamp.png" width="240"/> <img alt="lamp geometry" height="135" src="images/lamp_init.png" width="240"/> <img alt="lamp style" height="135" src="images/lamp_final.png" width="240"/> </p> <p float="center"> <img alt="horse" height="135" src="images/horse.png" width="240"/> <img alt="horse geometry" height="135" src="images/horse_init.png" width="240"/> <img alt="horse style" height="135" src="images/horse_final.png" width="240"/> </p>Important tips for running on your own meshes
Text2Mesh learns to produce color and displacements over the input mesh vertices. The mesh triangulation effectively defines the resolution for the stylization. Therefore, it is important that the mesh triangles are small enough such that they can accurately potray the color and displacement. If a mesh contains large triangles, the stylization will not contain sufficent resolution (and leads to low quality results). For example, the triangles on the seat of the chair below are too large.
<p align="center"> <img alt="large-triangles" src="images/large-triangles.png" height="25%" width="25%" /> </p>You should remesh such shapes as a pre-process in to create smaller triangles which are uniformly dispersed over the surface. Our example remeshing script can be used with the following command (and then use the remeshed shape with Text2Mesh):
python3 remesh.py --obj_path [the mesh's path] --output_path [the full output path]
For example, to remesh a file name called chair.obj
, the following command should be run:
python3 remesh.py --obj_path chair.obj --output_path chair-remesh.obj
Other implementations
Kaggle Notebook (by neverix)
External projects using Text2Mesh
-
Endava 3D Asset Tool integrates Text2Mesh into their modeling software to create 3D assets for games.
-
Psychedelic Trips Art Gallery uses Text2Mesh to generate AI Art and fabricate (3D print) the results.
Citation
@InProceedings{Michel_2022_CVPR,
author = {Michel, Oscar and Bar-On, Roi and Liu, Richard and Benaim, Sagie and Hanocka, Rana},
title = {Text2Mesh: Text-Driven Neural Stylization for Meshes},
booktitle = {Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
month = {June},
year = {2022},
pages = {13492-13502}
}