代码目录:
关于项目流程,可以跳转我的这篇博文查看DeepLearing—CV系列(二十)——基于MTCNN与centerloss/arcloss实现的人脸识别项目流程,本篇博文在其基础上用Pytorch实现了人脸识别项目。
一、face.py
创建arcsoftmax分类器,创建人脸特征提取器(使用预训练模型)
import torchvision.models as models
import torch.nn as nn
import torch
import torch.nn.functional as F