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