【文件属性】：
文件名称：TransFG:这是PyTorch正式实施的“ TransFG”文件
文件大小：738KB
文件格式：ZIP
更新时间：2021-04-01 16:30:05
fine-grained-recognition Python
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模式下以
【文件预览】：
TransFG-master
----TransFG.png(733KB)
----requirements.txt(70B)
----models()
--------modeling.py(16KB)
--------configs.py(3KB)
----LICENSE(1KB)
----utils()
--------scheduler.py(3KB)
--------data_utils.py(7KB)
--------dist_util.py(711B)
--------__init__.py(0B)
--------dataset.py(23KB)
--------autoaugment.py(10KB)
----README.md(3KB)
----train.py(16KB)