Tensorflow implementation of Unsupervised learning of object landmarks by factorized spatial embeddings