Skip to content
Tensorflow implementation of Unsupervised learning of object landmarks by factorized spatial embeddings
Branch: master
Clone or download
Latest commit f4d63ff Feb 19, 2019
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
test_samples Delete K8M4.png May 18, 2018
utils The first version of the main file is created. May 18, 2018
LICENSE License file May 18, 2018
README.md Update README.md Feb 19, 2019
main.py Update main.py Jun 1, 2018

README.md

Factorized-Spatial-Embeddings

Tensorflow implementation of Unsupervised learning of object landmarks by factorized spatial embeddings by Thewlis el al. for unsupervised landmark detection.

Sample results

Test results on LFW with 8 landmarks (K=8, M=4), trained on CelebA dataset for 2 epochs. Test results on LFW with 16 landmarks (K=16, M=4), trained on CelebA dataset for 2 epochs.

Setup

Prerequisites

  • Tensorflow 1.4

Getting Started

First download the CelebA dataset or the UT Zappos50k shoes dataset, extract images and use them to train the model.

# clone this repo
https://github.com/alldbi/Factorized-Spatial-Embeddings.git
cd Factorized-Spatial-Embeddings
# train the model 
python main.py \
  --mode train \
  --input_dir (directory containing CelebA dataset) \ 
  --K 8  \ #number of landmarks to be learned

# test the model
python main.py \
  --mode test \
  --input_dir (directory containing testing images)
  --checkpoint (address of the trained model, which is /OUTPUT as default)
  --K 8
  

You can’t perform that action at this time.