facebookresearch / mae

PyTorch implementation of MAE https//arxiv.org/abs/2111.06377
Other
6.93k stars 1.17k forks source link

patchify and unpatchify #182

Open tingyushi opened 8 months ago

tingyushi commented 8 months ago

Hello,

Just a suggestion, maybe change the magic number "3" in patchify and unpatchify to in_chans, so that the model can better adapt grayscale images.

Thanks!!

limaodaxia commented 7 months ago

Hello,

Just a suggestion, maybe change the magic number "3" in patchify and unpatchify to in_chans, so that the model can better adapt grayscale images.

Thanks!!

hello, how to unpatchify?do you have the weights and bias of decoder?