search

IcarusWizard / MAE

PyTorch implementation of Masked Autoencoder
MIT License
234 stars 46 forks source link

how to make the masked patches not random #14

Closed wenhaowang1995 closed 1 year ago

wenhaowang1995 commented 1 year ago

Hi, is there any possible way to fix the position of masked patches? (for example. make all masked patches together into the middle of the image instead of spreading round the image randomly)

Thank you for your help!

IcarusWizard commented 1 year ago

Hi @wenhaowang1995. You can re-write the PatchShuffle class to implement a different masking scheme.

wenhaowang1995 commented 1 year ago

Hi @wenhaowang1995. You can re-write the PatchShuffle class to implement a different masking scheme.

oh, I see, thanks a lot!