← all repositories

pengzhiliang/MAE-pytorch

PyTorch implementation of Masked Autoencoders (MAE), a self-supervised vision transformer model for scalable visual learning.

MAE-pytorch
Not currently ranked — collecting fresh signals.
star history

This repository provides an unofficial PyTorch implementation of the MAE (Masked Autoencoders Are Scalable Vision Learners) paper. It implements the pretraining and fine-tuning processes for a vision transformer that learns by masking and reconstructing patches from images. The implementation includes shuffle/unshuffle operations for masked token handling and supports sine-cosine positional embeddings as described in the original paper.

Frequently asked

What is pengzhiliang/MAE-pytorch?
PyTorch implementation of Masked Autoencoders (MAE), a self-supervised vision transformer model for scalable visual learning.
Is MAE-pytorch open source?
Yes — pengzhiliang/MAE-pytorch is an open-source project tracked on heatdrop.
What language is MAE-pytorch written in?
pengzhiliang/MAE-pytorch is primarily written in Python.
How popular is MAE-pytorch?
pengzhiliang/MAE-pytorch has 2.7k stars on GitHub.
Where can I find MAE-pytorch?
pengzhiliang/MAE-pytorch is on GitHub at https://github.com/pengzhiliang/MAE-pytorch.

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