Alpha-VL / ConvMAE

ConvMAE: Masked Convolution Meets Masked Autoencoders
MIT License
477 stars 41 forks source link

image unpatchify related problems #28

Open pengjuRen99 opened 1 year ago

pengjuRen99 commented 1 year ago

There is a little problem in the open source code, self.patch_embed is not defined in the unpatchify function of the model. The original dimension of the image cannot be restored. I hope it can be modified slightly for our convenience. Thank you for your answer.

pengjuRen99 commented 1 year ago

Also change p to 16 in unpatchify and patchify.

gaopengpjlab commented 1 year ago

Thanks for your suggestion. We will modify the code according to your suggestion.