Spaces:
Runtime error
Runtime error
File size: 625 Bytes
23a9607 d3361e8 23a9607 d7e38aa |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
---
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.). |