# FaceRecognition

**Repository Path**: zbzbzzz/FaceRecognition

## Basic Information

- **Project Name**: FaceRecognition
- **Description**: No description available
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 4
- **Forks**: 5
- **Created**: 2021-05-04
- **Last Updated**: 2024-06-01

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# 简介

本项目基于pytorch+python3.8 实现比较简单用于深度学习入门

由yolo实现人脸检测，resnet+arcface实现人脸识别。

项目中不包含训练代码只有检测代码，已训练好的权重已放在weights文件夹想下可以直接使用。

由于时间限制yolo用的是最小的yolov5s模型，resnet+arcface也没训练多久，所以可能准确率不高。



# 运行

第一步是要先生成人物的特征

```
python gen_features.py 
```

- **--source**	指定要生成人物特征的照片目录，一般是文件名对应这个特征的标签，比如xxx.jpg的标签就是xxx
- **--yolo_weight**    指定已经训练好的yolo权重
- **--arcface_weight**    指定已经训练好的resnet+arcface权重
- **--out_dir**    指定人物特征的输出目录，可以理解为人物特征数据库



生成完特征就可以根据已有的特征去识别人物了

```
python detect.py 
```

- **--source**    指定要检测的目录，值为0时打开摄像头检测
- **--yolo_weight**    指定已经训练好的yolo权重
- **--arcface_weight**    指定已经训练好的resnet+arcface权重

- **--person_dir**    指定已生成的人物特征的目录
- **--view-img**    开启摄像头要启用这个参数，显示图片



# 训练

在yolov5官方库做了点修改

https://gitee.com/zbzbzzz/yolov5



resnet+arcface在github开源项目中做调整后训练

https://gitee.com/zbzbzzz/arcface-pytorch



# 参考

https://github.com/ultralytics/yolov5

https://github.com/wudashuo/yolov5

https://github.com/ronghuaiyang/arcface-pytorch

https://github.com/ooooxianyu/yoloV5-arcface_forlearn