This is the implementation of the paper "Relational Neural Expectation Maximization: Unsupervised Discovery of Objects and their Interactions" in PyTorch. The code structure has been inspired form the original implementation in Tensorflow.
This is the implementation of the paper "Relational Neural Expectation Maximization: Unsupervised Discovery of Objects and their Interactions" in PyTorch. The code structure has been inspired form the original implementation in Tensorflow.