kaanaksit / odak

Scientific computing library for optics, computer graphics and visual perception.
https://kaanaksit.com/odak
Mozilla Public License 2.0
176 stars 52 forks source link

self-attention for CV #64

Closed aguzel closed 1 year ago

aguzel commented 1 year ago

I added the self-attention layer for Convolutional networks by using : "Non-local Neural Networks" (https://arxiv.org/abs/1711.07971)

aguzel commented 1 year ago

indentation fixed