YOLO v3、FaceNet和SVM的人脸检测识别系统源码(python)分享

createh52周前 (02-04)技术教程16

先转发,然后关注@大数据java架构师 私信“人脸”获取下载链接

项目简介

fine-tune YOLO v3 + FaceNet进行人脸识别,辨别。

1. 项目结构

--data

--|------baseface 图片、根据这些图片训练的128d向量,以及文件夹与人名的映射文件

---------|------0 第一个人的图片tag=0

---------|------1 第二个人的图片tag=1

......

---------|------n 第n个人的图片tag=n

---------|------map.txt 文件夹与人名的映射,依次放即可

---------|------vector.csv 根据这些图片得到的128维向量以及其类别(文件夹名)

--|------weights_facenet 模型文件(facenet)

--|------weights_yolo 模型(fune-tuning的yolo v3)

--|------weights_svm 模型(根据csv文件训练的svm模型)

--|------face-names 预测类别,默认即可

--|------yolo_anchors.txt 训练yolo v3时聚类得到的anchors框

--net yolo和facenet的网络

--preprocessing 预处理工具,下文的使用方法前期步骤都需要在这里运行

--setting 模型参数,可根据需要修改

--utils 封装的一些方法

--test.py 主体方法

2.模型文件(权重)

先转发,然后关注@大数据java架构师 私信“人脸”获取

3.yolo v3

YOLO v3的详细预测和训练,可到本人另一个项目YOLO_v3_tensorflow了解。

4.踩的坑

  1. fine-tune yolo v3时使用的是wider face数据集,其中有两个标注框是宽度或高度为0的,错误标注,筛选的时候需要去除掉这两个标注框,否则nms会报除0的异常。
  2. 训练svm时,需要标准化,预测时需要用同参数标准化再预测,否则svm预测结果都相同。
  3. 之前使用model-20170512-110547模型,输出128d向量,效果不好,换成model-20180402-114759模型,输出512d向量,效果有所提升。
  4. 每个人15张脸部图片,训练svm效果一般,可以增大样本量或者使用一些svm的tricks。

三、使用方法

项目需要安装以下包

numpy
pandas
opencv-python
scikit-learn
tensorflow
pillow

1.放置模型文件

下载yolo_face和facenet模型文件,放置在data文件夹下,如图

2.制作自己的人脸数据集

截取需要识别的人物的脸部图片,一人一个文件夹,文件夹名称从0开始依次累加,放在路径data/base_face下。

并修改data/map.txt,以空格分隔,下标和人脸文件夹名一一对应(map.txt中第一条下标为0,对应data/base_face/0文件夹)。

3.使用工具将图片转换成向量并存储

运行preprecessing/pre_tools.py内的save_vector_csv(),自动将图片使用facenet转换为128d或512d向量,并存储为data/base_face/vector.csv

4.训练svm分类器

基于已经储存的vector.csv文件,进行标准化后,运行preprecessing/pre_tools.py内的train_face_svm(),使用scikit-learn训练svm模型,并储存在data/weights_svm/svm.pkl中。

5.开始测试

5.1图片测试

放在data/test_img下,将setting/yolo_args.py中的detect_object改为imginput_image 改为图片路径。

运行根目录下的test.py文件。

另外如果要存储检测后的图片,将setting/yolo_args.py中的output_image改为要存储的路径,并保证各层级文件夹存在

5.2视频测试

放在data/test_video下,将setting/yolo_args.py中的detect_object改为videoinput_video 改为视频路径。

运行根目录下的test.py文件。

另外如果要存储检测后的视频,将setting/yolo_args.py中的output_video改为要存储的路径,并保证各层级文件夹存在

先转发,然后关注@大数据java架构师 私信“人脸”获取下载链接

相关文章

Java 和.NET,谁将“统治”未来(java和netbeans)

随着时代的变化,技术在不断发展,编程语言的重要性也在发生变化。所以,要我们在两门语言中二选一,通常很困难。在这篇文章里,我们要讨论的是 Java 和.NET 之间的“战争”。.NET 和 Java 是...

JAVA和.NET谁更优秀?你选择谁?(java和.net哪个简单)

Java和.NET都是现代流行的编程语言,它们在许多方面都有各自的优缺点和优势。下面我们将详细分析Java和.NET谁更优秀,并给出具体理由。性能:Java和.NET在性能上没有明显的区别。它们都是基...

为什么说微软开源 .Net 也拼不过 Java?

.Net 服务器端源代码的开源对于微软而言是很重大的一步。然而这个动作的影响有限,也必须说 .Net 距离追上 Java 的开放源代码可能还要很长一段时间。 关于 .Net 开放源代码的计划,微软副总...

.NET Core/.NET5/.NET6 开源项目:工作流组件

前言开源项目是众多组织与个人分享的组件或项目,作者付出的心血我们是无法体会的,所以首先大家要心存感激、尊重。请严格遵守每个项目的开源协议后再使用。尊重知识产权,共建和谐开源社区。ELSAElsa Co...

谁说.NET没有GC调优,只改一行代码就让程序不再占用内存

经常看到有群友调侃“为什么搞Java的总在学习JVM调优?那是因为Java烂!我们.NET就不需要搞这些!”真的是这样吗?今天我就用一个案例来分析一下。昨天,一位学生问了我一个问题:他建了一个默认的A...

Avalonia:可信创.NET 跨平台UI,让JAVA失业者转.NET信创开发!

亲爱的读者们,今天我想与大家分享一个令人兴奋的主题 —— Avalonia,这个强大的.NET跨平台UI框架。作为一名曾经的JAVA开发者,我深知转换技术栈的挑战。然而,在当前快速变化的IT行业中,适...