Face-Transformer:人脸识别变压器-源码

脸部变形金刚 这是用于识别的人脸变压器（ ）的代码。 最近，不仅在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 将