masked-image-modeling 开源项目推荐（61 个）

open-mmlab /mmpretrain

OpenMMPLab预培训收件箱和基准

open-mmlab /mmselfsup

OpenMMLab自我监督学习工作表和基准

Robbyant /lingbot-depth

空间感知的掩蔽深度建模

keyu-tian /SparK

[ICLR'23 Spotlight🔥] The first successful BERT/MAE-style pretraining on any convolutional network; Pytorch impl. of "Designing BERT for Convolutional Networks: Sparse and Hierarchical Masked Modeling"

microsoft /SimMIM

This is an official implementation for "SimMIM: A Simple Framework for Masked Image Modeling".

Westlake-AI /openmixup

CAIRI监督半监督及自监督视觉表现学习工具箱与基准测试

Alpha-VL /ConvMAE

ConvMAE: Masked Convolution Meets Masked Autoencoders

Lupin1998 /Awesome-MIM

[Survey] Masked Modeling for Self-supervised Representation Learning on Vision and Beyond (https://arxiv.org/abs/2401.00897)

hustvl /MIMDet

[ICCV 2023] You Only Look at One Partial Sequence

lxtGH /CAE

This is a PyTorch implementation of “Context AutoEncoder for Self-Supervised Representation Learning"

LayneH /GreenMIM

[NeurIPS2022] Official implementation of the paper 'Green Hierarchical Vision Transformer for Masked Image Modeling'.

Sense-X /MixMIM

MixMIM: Mixed and Masked Image Modeling for Efficient Visual Representation Learning

JIA-Lab-research /MOOD

Official PyTorch implementation of MOOD series: (1) MOODv1: Rethinking Out-of-distributionDetection: Masked Image Modeling Is All You Need. (2) MOODv2: Masked Image Modeling for Out-of-Distribution Detection.

Atten4Vis /CAE

This is a PyTorch implementation of “Context AutoEncoder for Self-Supervised Representation Learning"

Haochen-Wang409 /HPM

[CVPR'23 & TPAMI'25] Hard Patches Mining for Masked Image Modeling & Bootstrap Masked Visual Modeling via Hard Patch Mining

salesforce /MUST

PyTorch code for MUST

ariG23498 /mae-scalable-vision-learners

A TensorFlow 2.x implementation of Masked Autoencoders Are Scalable Vision Learners

liuxingbin /dbot

[ICLR2024] Exploring Target Representations for Masked Autoencoders

aicip /Cross-Scale-MAE

[NIPS'23] Official Code of the paper "Cross-Scale MAE: A Tale of Multi-Scale Exploitation in Remote Sensing"

yuxin-jiang /MRKD

This is an official implementation of “A Masked Reverse Knowledge Distillation Method Incorporating Global and Local Information for Image Anomaly Detection” (MRKD) with PyTorch, accepted by knowledge-based systems.