CV

This page contains the list of all attention models and non-local layers for computer vision available in Echo.

Echo contains implementation for following attention mechanisms for Computer Vision (✅ - implemented functions, 🕑 - functions to be implemented soon, ⬜ - function is implemented in the original deep learning package):

#	Function	PyTorch	TensorFlow	MegEngine
1	Triplet Attention	✅	🕑	🕑
2	CBAM	✅	✅	🕑
3	BAM	🕑	🕑	🕑
4	SE	✅	🕑	🕑