Skip to content

zhenjia-xu/psd

master
Switch branches/tags

Name already in use

A tag already exists with the provided branch name. Many Git commands accept both tag and branch names, so creating this branch may cause unexpected behavior. Are you sure you want to create this branch?
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Unsupervised Discovery of Parts, Structure, and Dynamics

This repository contains pre-trained models and testing code for PSD presented at ICLR 2019.

http://psd.csail.mit.edu

Teaser Image

Installation

Our current release has been tested on Ubuntu 18.04 LTS.

Cloning the repository

git clone git@github.com:zhenjia-xu/psd.git
cd psd

Set up Python environment:

pipenv install --dev

Guide

Generating shape dataset

pipenv run python data_generator.py

Downloading the pretrained model (185 MB)

./download_model.sh

Visualization

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.

Reference

@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).

About

[ICLR 2019] ]Unsupervised Discovery of Parts, Structure, and Dynamics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published