Skip to content

人脸全家桶--RetinaFace(MobileNetV2 and ResNet50 with Gender)、ArcFace、FaceBeautyRank and FaceRetrieval

Notifications You must be signed in to change notification settings

TalkUHulk/HulkFakeFace-pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

11 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

HulkFakeFace-pytorch

TalkUHulk

简介

HulkFakeFace为个人的人脸相关项目实践学习库,主要包括检测(retinaface)+识别(arcface)+相关应用(如颜值打分、检索),相关代码和预训练模型都可在本项目中找到。

🔥水平有限,有错误请及时指出,共同进步学习~

相关模型及数据集

链接: https://pan.baidu.com/s/1WDg6NwQODyll1Qour5GeAQ 密码: tw16

环境

  • python3
  • torch==1.6.0
  • torchvision==0.7.0
  • opencv_contrib_python==4.4.0.44
  • warmup_scheduler==0.3.2
  • tensorboardX==2.1
  • pymilvus==0.2.14

RetinaFace

参考项目

  • backbone: mobilenet-v2 | resnet50
  • datasets: widerface
  • addition: 在原始网络中加入性别判断,目前仅在resnet50中支持,cfg_re50.gender=True
  • tensorboard:

logs

  • FDDB ROC:

ROC

  • demo:

girls

nba

stars

GenderClassify

  • 简介: 主要用来标注widerface,为RetinaFace with gender制作训练集

ArcFace

  • Pretrained Models: InsightFace_Pytorch 的基础上,使用亚洲人脸fine-tune
  • Performance:
LFW(%)
0.9943
  • tensorboard:
    Loss

Arc

Beauty

  • loss: train with arcface and pretrained with MarginRankingLoss
  • datasets: SCUT_FBP5500
  • Performance:
SCUT_FBP5500-MAE
0.277
  • tensorboard:
    Train

Val

Retrieval

Thanks to dataset provider:Copyright(c) 2018, seeprettyface.com, BUPT_GWY contributes the dataset.

ab

st

tly

References

About

人脸全家桶--RetinaFace(MobileNetV2 and ResNet50 with Gender)、ArcFace、FaceBeautyRank and FaceRetrieval

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages