This directory is designed for training a SD classifier. Below is the directories structure and how to run
code/
├── AutoEncoder/ # Files for our autoencoder architecture and training
├── Data/ # Where to place data
├── Helper/ # Where helper functions and data prossessing code lives
├── report_assets/ # Figures and data from training
├── SignDistanceModel/ # Files for our sign distance model architecture and training
└── README.md # Repository overview (this file)
-
Create a Data folder in the root of the code directory
-
Download the ModelNet40 dataset and place in Data folder
-
Run the repair script on the data
python3 Helpers/repair_dataset.py
-
Train the autoencoder model
python3 AutoEncoder/train.py
-
Download the SDF dataset from our google drive and place it in the Data folder
-
Train the SDF classification model
python3 SignDistanceModel/train.py