/learnable-deep-priors

Code for the paper "Spatial Mixture Models with Learnable Deep Priors for Perceptual Grouping"

Primary LanguagePython

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.