文件名称：TransFG:这是PyTorch正式实施的“ TransFG”文件
文件大小：738KB
文件格式：ZIP
更新时间：2024-04-21 10:16:45
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)