TNT

论文：Transformer in Transformer
官方项目：huawei-noah/noah-research
模型代码：tnt.py

验证集数据处理：

# 图像后端：pil
# 输入图像大小：224x224
transforms = T.Compose([
    T.Resize(248, interpolation='bicubic'),
    T.CenterCrop(224),
    T.ToTensor(),
    T.Normalize(mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5])
])

模型细节：

Model Model Name Params (M) FLOPs (G) Top-1 (%) Top-5 (%) Pretrained Model

TNT-S tnt_s 23.8 5.2 81.53 95.74 Download

引用：

@misc{han2021transformer,
    title={Transformer in Transformer}, 
    author={Kai Han and An Xiao and Enhua Wu and Jianyuan Guo and Chunjing Xu and Yunhe Wang},
    year={2021},
    eprint={2103.00112},
    archivePrefix={arXiv},
    primaryClass={cs.CV}
}