ViT Attention Map Visualization
ViT Attention Map을 Visualization할 일이 있어서 기록을 합니다.환경: Jupyter-lab모델: timm's ViT-base Import libsimport torchimport timmfrom torchvision import transformsfrom PIL import Imageimport matplotlib.pyplot as pltimport numpy as npimport requests Load Imagepreprocess = transforms.Compose([ transforms.Resize((224, 224)), transforms.ToTensor(), transforms.Normalize( mean=(0.5, 0.5, 0.5), ..
