传输网
此存储库包含代码
用法
1.下载Google预先训练的ViT模型
:R50-ViT-B_16,ViT-B_16,ViT-L_16 ...
wget https://storage.googleapis.com/vit_models/imagenet21k/{MODEL_NAME}.npz &&
mkdir ../model/vit_checkpoint/imagenet21k &&
mv {MODEL_NAME}.npz ../model/vit_checkpoint/imagenet21k/{MODEL_NAME}.npz
2.准备数据
请访问以获取详细信息,或将电子邮件发送至jienengchen01 AT gmail.com以请求预处理的数据。 如果您想使用预处理后的数据,请将其用于研究目的,请勿重新分发。
3.环境
请准备一个python = 3.7的环境,然后使用命令“
2021-04-01 20:39:51
26KB
Python
1