README.md

metadata

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). You'd just need to adjust the code accordingly.

CaiT was proposed in Going deeper with Image Transformers (Touvron et al.).