search

SudeepDasari / data4robotics

MIT License
56 stars 9 forks source link

Cannot reconstruct the image with released pre-trained features #5

Closed ShutongJIN closed 4 weeks ago

ShutongJIN commented 2 months ago

Hi,

Thanks for sharing the code! I ran into an issue when trying to reconstruct the image using the released pre-trained weights(vit_base/IN_1M.pth and SOUP_1M_IN.pth). The reconstruction visualization code+hyperparameters follow the original code in MAE: https://github.com/facebookresearch/mae/blob/main/demo/mae_visualize.ipynb

An example of the reconstruction figure(from ImageNet) is attached. failed_reconstruction

Thanks!