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.
Inference Providers
NEW
This model is not currently available via any of the supported Inference Providers.
The model cannot be deployed to the HF Inference API:
The model has no library tag.