何愷明MAE,開源了!!
大家好,我是DASOU。
之前我在B站錄制了一起講解MAE算法的視頻,播放量還不錯,感興趣的可以去看一眼。
MAE做的一個事情簡單來說,就是通過隨機遮擋,進行圖像重建,進而強迫模型學(xué)習(xí)知識。
并且遮擋的比例,非常大!我們直接看效果:

第一列是遮擋圖,第二列是MAE模型重建結(jié)果,第三列是原圖。
這個效果是相當(dāng)?shù)捏@艷的。
前幾天發(fā)現(xiàn)官方代碼開源了MAE,目前是1.8k的星標(biāo):

?Visualization demo
?Pre-trained checkpoints + fine-tuning code
?Pre-training code
對MAE感興趣或者自己項目中會用到的朋友可以自己去跑一下,代碼地址在這里:
https://github.com/facebookresearch/mae
評論
圖片
表情
