Commit Graph

2 Commits (68129f0a9c84d89057429116021d0544091bb12a)

Author SHA1 Message Date
Yixiao Fang 73cd764b5f
[Feature] Support pixel reconstruction visualization (#570)
* refactor reconstruction visualization

* support simmim visualization

* fix reconstruction bug of MAE

* support visualization of MaskFeat

* refaction mae visualization demo

* add unit test

* fix lint and ut

* update

* add docs

* set random seed

* update

* update docstring

* add torch version check

* update

* rename

* update version

* update

* fix lint

* add docstring

* update docs
2022-12-06 19:45:01 +08:00
RenQin 41747f73c7 [Refactor]: refactor MAE visualization (#471)
* [Refactor]: refactor MAE visualization

* [Fix]: fix lint

* [Refactor]: refactor MAE visualization

* [Feature]: add mae_visualization.py

* [UT]: add unit test

* [Refactor]: move mae_visualization.py to tools/analysis_tools

* [Docs]: Add the purpose of the function unpatchify()

* [Fix]: fix lint
2022-11-03 16:09:36 +08:00