File size: 629 Bytes
31877e7 37a8469 31877e7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
---
license: mit
---
# 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](https://sites.research.google/trc/about/) to train these models. |