search

facebookresearch / dinov2

PyTorch code and models for the DINOv2 self-supervised learning method.
Apache License 2.0
8.74k stars 751 forks source link

Attention maps Visualization #294

Open Rukhsar111 opened 10 months ago

Rukhsar111 commented 10 months ago

I've trained a Image classification model using Dinov2 for my custom Use case . Now I want generate the Attention maps for my trained model . How can I do that?

AriesChen-UPC commented 10 months ago

Using https://github.com/facebookresearch/dinov2/issues/90, you can generate attention maps of the dinov2 version, but it is unable to generate attention maps of the dinov2 with registers version.