TransFG:这是PyTorch正式实施的“ TransFG”文件-源码

TransFG：用于细粒度识别的变压器体系结构 论文的官方PyTorch代码： 不久将发布基于ImageNet-1K上训练的DeiT的实现，并进行精调。 框架 依存关系： 的Python 3.7.3 PyTorch 1.5.1 火炬视觉0.6.1 ml_collections 用法 1.下载Google预先训练的ViT模型 ：ViT-B_16，ViT-B_32 ... wget https://storage.googleapis.com/vit_models/imagenet21k/{MODEL_NAME}.npz 2.准备数据 在本文中，我们使用来自5个公开数据集的数据： 请从官方网站下载它们，并将它们放在相应的文件夹中。 3.安装所需的软件包 使用以下命令安装依赖项： pip3 install -r requirements.txt 4.火车 要在FP-16模式下以