脸部变形金刚
这是用于识别的人脸变压器( )的代码。
最近,不仅在NLP中,而且在计算机视觉中,都对Transformer产生了极大的兴趣。 我们想知道是否可以将变压器用于人脸识别,以及它是否比CNN更好。 因此,我们研究了人脸识别中Transformer模型的性能。 这些模型在大规模人脸识别数据库MS-Celeb-1M上进行了训练,并在几个主流基准上进行了评估,包括LFW,SLLFW,CALFW,CPLFW,TALFW,CFP-FP,AGEDB和IJB-C数据库。 我们证明,在具有相似数量的参数和MAC的情况下,Transformer模型可实现与CNN相当的性能。
使用说明
1.准备
该代码主要从和。 除了PyTorch和torchvision,安装由,和包由。 衷心感谢他们的贡献。
pip install vit-pytorch
pip install timm==0.3.2
将
2021-10-26 20:27:26
171KB
Python
1