PyTorch - MAE(Masked Autoencoders)推理脚本

Posted 2022-11-21 SpikeKing

MAE推理脚本：

• 需要安装：pip install timm==0.4.5
• 需要下载：mae_visualize_vit_base.pth，447M

源码：

#!/usr/bin/env python
# -- coding: utf-8 --
"""
Copyright (c) 2022. All rights reserved.
Created by C. L. Wang on 2022/10/21
"""

import os

import matplotlib.pyplot as plt
import numpy as np
import torch