Added Triplet Attention Mechanism

lucasjinreal / nb

Neural Network Blocks - Collect all kinds of fancy model blocks for you to build more powerful neural network model.

Other

237 stars 53 forks source link

Added Triplet Attention Mechanism #5

Closed digantamisra98 closed 3 years ago

digantamisra98 commented 3 years ago

Reference - https://github.com/LandskapeAI/triplet-attention (WACV 2021) Paper - https://openaccess.thecvf.com/content/WACV2021/html/Misra_Rotate_to_Attend_Convolutional_Triplet_Attention_Module_WACV_2021_paper.html

lucasjinreal commented 3 years ago

@digantamisra98 thanks for PR, could u also write an updates in README.md indicates users how to call this block? With an example tensor and showing what the shape we expected from this block.

digantamisra98 commented 3 years ago

@jinfagang updated Readme

lucasjinreal commented 3 years ago

@digantamisra98 Thank u! LGTM.