|
--- |
|
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. |
