/scs_pytorch

PyTorch implementation of 2D Sharpened Cosine Similarity layer

Primary LanguagePython

scs_pytorch

Pytorch implementation of the Sharpened Cosine Similarity layer mentioned in this tweet.

The current version supports square kernels with arbitrary kernel sizes.

Part of the implementation borrows from dvisockas/cos_sim.

To do

  • depth-wise conv (with groups arguement)
  • non-square kernels
  • efficiency
  • demo training code