Invariant Slot Attention: Object Discovery with Slot-Centric Reference Frames
This is a model repository for my paper Invariant Slot Attention. I uploaded a single checkpoint for each dataset + model variant I used in the paper except for Waymo Open trained with depth targets.
Code (JAX/FLAX): https://github.com/google-research/google-research/tree/master/invariant_slot_attention
Paper (ICML'23): https://arxiv.org/abs/2302.04973
Demo: https://huggingface.co/spaces/ondrejbiza/isa
Acknowledgements
I used the TPU Research Cloud to train these models.