AutoEncoder
ShapeNet
Download the dataset for training and testing here. Clone the latest
ocnn-pytorchrepository, and enter the subdirectoryprojects. Unzip and place the data in the folderdata/ShapeNetV1.Run the following command to train the network. The training log and weights can be downloaded here.
python autoencoder.py --config configs/ae_shapenet.yaml
Run the following command to get the predictions on the testing dataest. The parameter following
SOLVER.ckptcan be freely modified to test different trained weights. And the results are in the folderlogs/ae_shapenet/ae_eval.python autoencoder.py --config configs/ae_shapenet.yaml \ SOLVER.run evaluate SOLVER.alias eval \ SOLVER.ckpt logs/ae_shapenet/ae/checkpoints/00300.model.pth