3D mesh stylization driven by a text input in PyTorch
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” (forthcoming).
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
System requirements
System Requirements
- Python 3.7
- CUDA 10.2
- 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
Citation
@article{text2mesh,
author = {Michel, Oscar
and Bar-On, Roi
and Liu, Richard
and Benaim, Sagie
and Hanocka, Rana
},
title = {Text2Mesh: Text-Driven Neural Stylization for Meshes},
journal = {arXiv preprint arXiv:2112.03221},
year = {2021}
}