This repository contains pre-trained models and testing code for PSD presented at ICLR 2019.
Our current release has been tested on Ubuntu 18.04 LTS.
git clone git@github.com:zhenjia-xu/psd.git
cd psd
pipenv install --dev
pipenv run python data_generator.py
./download_model.sh
pipenv run python demo.py --gpu ID
- --gpu ID: use which gpu (starting from 0). Set to -1 to use CPU only.
The results will be presented in ./demo/index.html
.
@inproceedings{psd,
title={Unsupervised Discovery of Parts, Structure, and Dynamics},
author={Xu, Zhenjia and Liu, Zhijian and Sun, Chen and Murphy, Kevin and
Freeman, William T and Tenenbaum, Joshua B and Wu, Jiajun},
booktitle={International Conference on Learning Representations (ICLR)},
year={2019},
}
For any questions, please contact Zhenjia Xu (xuzhenjia@cs.columbia.edu).