spencerhhubert / chisel

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

chisel

CAD Copilot - Generate meshes/STLs from textual prompts

Chisel uses a similar architecture to Stable Diffusion where a transformer UNet neural network is used to learn to undo gaussian noise added an image, except instead of RGB values in an image, we consider the 3D coordinate values that comprise a 3D mesh.

A mesh, often seen in the STL file format, are 3D objects defined by listed of vertices connected by faces, that are themselves each defined in terms of three vertices. This is equivalent to a hypergraph where each hyperedge is a face, and we use PyTorch Geometric for learning on these hypergraphs. We'll replicate the transformer UNet architecture with concepts from graph attention network, GNNs for convolutions, and top k pooling for the downsamples.

Undoing the noise on a 3D mesh

*this is not a product of the model, this is an manual algorithmically generated example of how it works

The initial training is being completed with the ABC Dataset.

About


Languages

Language:Python 67.1%Language:Shell 28.6%Language:Dockerfile 4.3%