文件名称:Masked Autoencoders Are Scalable Vision Learners (MAE)代码样例
文件大小:228KB
文件格式:ZIP
更新时间:2022-05-27 22:28:53
源码软件
解压后得到test文件夹,在下面的网址中下载预训练后的模型 https://dl.fbaipublicfiles.com/mae/visualize/mae_visualize_vit_large.pth 将其移动到test/demo中,进入demo文件夹,运行demo.py文件即可
【文件预览】:
test
----engine_pretrain.py(3KB)
----main_linprobe.py(13KB)
----main_finetune.py(15KB)
----engine_finetune.py(5KB)
----submitit_finetune.py(4KB)
----submitit_linprobe.py(4KB)
----__pycache__()
--------models_mae.cpython-36.pyc(7KB)
----submitit_pretrain.py(4KB)
----models_vit.py(2KB)
----models_mae.py(10KB)
----util()
--------misc.py(11KB)
--------pos_embed.py(4KB)
--------crop.py(1KB)
--------datasets.py(2KB)
--------__pycache__()
--------lars.py(2KB)
--------lr_sched.py(802B)
--------lr_decay.py(2KB)
----main_pretrain.py(9KB)
----demo()
--------masked_mk.png(55KB)
--------mk.png(131KB)
--------demo.py(4KB)