Skip to content

This is an unofficial PyTorch 1.0.1 implementation of the papr Neural Aggregation Network for Video Face Recognition. CVPR 2017

Notifications You must be signed in to change notification settings

GuardSkill/NAN_pytorch

Repository files navigation

#face_rec NAN 人脸识别复现(现只在YouTube Face 数据集上)

安装环境

  • python版本:3.6
  • pytorch版本:0.4.1
  • Jupyter(可选)

任务

1.    论文的复现,数据集的划分 (:heavy_check_mark: Accruacy:) 95.5% for Youtube Dataset
2.    光流数据加入  (:clock130:)
3.    额外数据集训练实验 (:clock430:)

文件介绍

  • Dataset.py:负责训练数据的读取
  • Inception.py:Googlenet的架构文件
  • Network.py:NAN网络结构以及一些子网络结构
  • ShowData.ipynb:展示数据和测试Dataset.py的jupyter文件
  • TrainCNN.ipynb:训练CNN网络的jupyter文件(可用来测试训练逻辑)
  • Train.ipynb/train_ver.py:训练网络的已经测试训练逻辑的jupyter文件
  • TrainVerificationTask.ipynb:训练NAN的聚合模块的jupyter文件
  • util.py: 用于计算人脸验证中的TPR,FPR率,用于绘制ROC曲线

版本

  • V2.5 更新时间:2019-9-27 (设置聚合模块训练帧数为100帧,达到论文复现效果)
  • V2.0 更新时间:2019-9-26 (加入NAN模型的训练,ROC曲线的绘制,精度的计算)
  • V1.5 更新时间:2019-9-25 (加入CNN模型的单独训练)
  • V1.0 更新时间:2019-9-22 (改动大部分代码,初始化库)

数据集:

Youtube Face
http://www.cslab.openu.ac.il/download/wolftau/YouTubeFaces.tar.gz http://www.cslab.openu.ac.il/download/wolftau/frame_images_DB.tar.gz

GuardSKill

About

This is an unofficial PyTorch 1.0.1 implementation of the papr Neural Aggregation Network for Video Face Recognition. CVPR 2017

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published