README.md

shapenet

This is a fork from original repo shapenet, which contains reference, paper and explanation.

Changes from original repo

Changes are mostly about preprocessing training data. Images are saved as numpy binary for faster subsequent execution. And some dependencies are removed so that it is simpler to run.

Download training data

Data can be downloaded from here

Available command

To make pca

cd shapenet/scripts && python make_pca.py

To preprocess data

python -m shapenet.scripts.preprocess

To train

python -m shapenet.scripts.train --datadir "/path/to/data"

To predict

python -m shapnet.scripts.predict --datadir "/path/to/data" --img "/path/to/target/image"