Spaces:
Runtime error
Runtime error
| title: Class Attention Map | |
| emoji: 🤖 | |
| colorFrom: red | |
| colorTo: purple | |
| sdk: gradio | |
| sdk_version: 2.9.4 | |
| app_file: app.py | |
| pinned: false | |
| license: apache-2.0 | |
| This Space shows the spatial-class relationship of a CaiT model by plotting its | |
| class attention map w.r.t an input image. For speed, this Space uses the smallest | |
| CaiT model (`cait_xxs24_224`) but you can try it for others too (model collection | |
| if [here](https://tfhub.dev/sayakpaul/collections/cait/1)). You'd just need to | |
| adjust the code accordingly. | |
| CaiT was proposed in [Going deeper with Image Transformers](https://arxiv.org/abs/2103.17239) | |
| (Touvron et al.). |