/slot-attention-pytorch

Pytorch Implementation of paper "Object-Centric Learning with Slot Attention"

Primary LanguageJupyter Notebook

slot-attention-pytorch

This repo is the Pytorch Implementation of paper "Object-Centric Learning with Slot Attention". It uses the slot attention module from here but augments it with encoder, decoder and train/test scripts.

The official Tensorflow repository has been released here. This repo should be a direct translation from the official Tensorflow repo to a PyTorch version.

Citation

@misc{locatello2020objectcentric,
    title = {Object-Centric Learning with Slot Attention},
    author = {Francesco Locatello and Dirk Weissenborn and Thomas Unterthiner and Aravindh Mahendran and Georg Heigold and Jakob Uszkoreit and Alexey Dosovitskiy and Thomas Kipf},
    year = {2020},
    eprint = {2006.15055},
    archivePrefix = {arXiv},
    primaryClass = {cs.LG}
}