Spatial Mixture Models with Learnable Deep Priors (SMMLDP)

This is the code repository of the paper "Spatial Mixture Models with Learnable Deep Priors for Perceptual Grouping".

Dependencies

• pytorch == 1.8
• torchvision == 0.9
• numpy == 1.19
• h5py == 2.10
• pyyaml == 5.4
• scipy == 1.6
• scikit-learn == 0.24

Datasets

Change the current working directory to data and run create_datasets.sh.

cd data
./create_datasets.sh
cd ..

Experiments

Change the current working directory to experiments and run run.sh.

cd experiments
./run.sh
cd ..

Run experiments/evaluate.ipynb to evaluate the trained models.