class-attention-map / README.md
sayakpaul's picture
sayakpaul HF staff
Update README.md
d3361e8
|
raw
history blame contribute delete
No virus
625 Bytes

A newer version of the Gradio SDK is available: 5.0.0

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