Below are instructions on how to setup the completion3d data loader. New datasets can be added following the template in shared/datasets/shapenet.py. Feel free to submit new dataset additions to the benchmark as a pull request.
git clone [email protected]:lynetcha/completion3d.git
cd completion3d
PYTHON_BIN=/path/to/python3.6
virtualenv -p $PYTHON_BIN comp3d_venv
source comp3d_venv/bin/activate
pip install -r requirements/data-requirements.txt
Download the completion3D dataset. You should have the following direcory structure
data
├── shapenet
│ ├── synsetoffset2category.txt
│ ├── train
│ ├── val
│ ├── test
│ ├── train.list
│ └── val.list
│ └── test.list
python shared/datasets/shapenet.py