← all repositories

landskape-ai/triplet-attention

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

440 stars Jupyter Notebook Computer VisionML Frameworks
triplet-attention
Velocity · 7d
+0.2
★ / day
Trend
steady
star history

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.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.