landskape-ai/triplet-attention
PyTorch implementation of triplet attention, a novel cross-dimensional attention module for convolutional neural networks.

This repository provides the official implementation of the Triplet Attention module from the WACV 2021 paper. The method uses a three-branch structure with rotation operations to capture inter-dimensional dependencies in feature tensors, improving channel-spatial interaction without significant computational overhead. The module is designed as a drop-in enhancement for CNNs and evaluates on ImageNet classification and object detection tasks.