Image2Shape

ShapeNet

Download the dataset for training and testing here. Clone the latest ocnn-pytorch repository, and enter the subdirectory projects. Unzip and place the data in the folder data/ShapeNetV1.
Run the following command to train the network. The training log and weights can be downloaded here. The training process takes 9 hours on 4 Nvidia 2080 GPUs.
```
python image2shape.py --config configs/image2shape.yaml SOLVER.gpu 0,1,2,3
```
Run the following command to get the predictions on the testing dataest. The parameter following SOLVER.ckpt can be freely modified to test different trained weights. And the results are in the folder logs/image2shape/eval.
```
python image2shape.py --config configs/image2shape.yaml             \
       SOLVER.run evaluate  SOLVER.alias eval                       \
       SOLVER.ckpt logs/image2shape/image2shape/checkpoints/00300.model.pth
```