The repository adds a small wrapper on the marching-primitives codebase released by the authors for the Visual Shape Inference Challenge.
- Download the dataset from 3DCoMPaT repository. Unzip the 3DCoMPaT dataset to get the
.gtlf
mesh files. - Download the
vsic_split.json
from the vsic repository. - Run the
run_mps.py
script with the relevant args (or change theDATA_DIR
andSPLIT_FILE
variables on the top of the script). This will run the system on all the shapes in thetest
split. The output will be saved asfinal_expressions.pkl
. This file can be submitted to the evaluation server directly. - You can also launch multiple jobs with
run_mps_parallel.py --n-jobs 12 --proc-id n
. This will save multiple filesfinal_expressions_{proc-id}.pkl
which need to be combined together to create the final submission file.