search

BobMcDear / attention-in-vision

PyTorch implementation of popular attention mechanisms in vision
https://bobmcdear.github.io/posts/attention-in-vision/
15 stars 2 forks source link

Wrapper Code #1

Closed karacanil closed 2 weeks ago

karacanil commented 1 month ago

Hey,

Great work with the repository. But a wrapper code that allows users to observe the implemented attention techniques in practice would be a great addition to this repository.

Have a great day.

BobMcDear commented 1 month ago

Hello,

Thank you for your interest. I am not sure what you mean by "observe the implemented attention techniques". Are you referring to a visualization toolbox for the attention mechanisms in this repository?

Best, Borna