← all repositories

The-AI-Summer/self-attention-cv

A PyTorch library providing self-attention mechanism implementations for computer vision applications.

1.2k stars Python ML FrameworksComputer Vision
self-attention-cv
Not currently ranked — collecting fresh signals.
star history

This repository provides implementations of various self-attention building blocks for computer vision tasks in PyTorch. It includes modules such as multi-head self-attention, axial attention, and transformer encoders, implemented using einsum and einops for efficient computation. The library is designed as reusable components that can be integrated into larger vision models.

Frequently asked

What is The-AI-Summer/self-attention-cv?
A PyTorch library providing self-attention mechanism implementations for computer vision applications.
Is self-attention-cv open source?
Yes — The-AI-Summer/self-attention-cv is open source, released under the MIT license.
What language is self-attention-cv written in?
The-AI-Summer/self-attention-cv is primarily written in Python.
How popular is self-attention-cv?
The-AI-Summer/self-attention-cv has 1.2k stars on GitHub.
Where can I find self-attention-cv?
The-AI-Summer/self-attention-cv is on GitHub at https://github.com/The-AI-Summer/self-attention-cv.

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