Reonstruction visualization for MAE series

open-mmlab / mmpretrain

OpenMMLab Pre-training Toolbox and Benchmark

https://mmpretrain.readthedocs.io/en/latest/

Apache License 2.0

3.39k stars 1.06k forks source link

Reonstruction visualization for MAE series #1823

Open apple629 opened 11 months ago

apple629 commented 11 months ago

Describe the feature

The model class doesn't work as demonstrated in https://colab.research.google.com/github/facebookresearch/mae/blob/main/demo/mae_visualize.ipynb

Will you implement it?

[ ] I would like to implement this feature and create a PR!

jackhu-bme commented 11 months ago

I also need this feature for comparing the reconstruction quality between MAE and MFF(ICCV2023), as the reconstruction results is not included in MFF paper, here is the link: https://github.com/open-mmlab/mmpretrain/tree/main/configs/mff. This is very essential for my study, need help. If possible, I need some suggestions for fast implementation of this visualization, as I'm not familiar with this repo.