Skip to content

Sunqinghong2333/stable-point-e

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

18 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Stable Point-E

We implemented the training code based on OpenAI/Point-E. In terms of data, we provide colored point cloud data from the ShapeNet dataset along with descriptive text. As for the model, we offer a checkpoint trained on the aforementioned training data.

Data Preparation

You need to download pcl.tar from GoogleDrive and extract it to the data/shapenet folder.

Usage

Install with pip install -e ..

To get started with aforementioned training data:

  • train.sh - training shell based on slurm, you can run as following: sh train.sh 32 train {your_slurm_partition} texts 5 point_e_save point_e/configs/shapenet.json

To visualize:

  • vis.sh - sample point clouds, conditioned on text prompts, you can run as following: sh vis.sh {your_slurm_partition}

For P-FID and P-IS evaluation scripts, see:

Acknowledgement

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published