pytorch实现人脸检测与识别_MTCNN人脸检测Pytorch实现

该项目实现了MTCNN人脸检测与识别,包括数据集的创建、模型训练和测试。数据集通过CNN_FacePoint和WIDER_Face进行标注,采用三种方式生成训练数据。训练了三个模型,方法二生成的图片虽小但数量庞大,方法三对CPU负担较大,而方法一的效果不佳。训练模型后,在detector.py中进行测试,当前模型效果有待优化。
摘要由CSDN通过智能技术生成

361de130cd29110865c8ea8d545db294.png

equation?tex=GitHub 项目地址:GitHberChen/MTCNN_Pytorch

接上篇:Uno Whoiam:MTCNN人脸检测:三个臭皮匠,顶个诸葛亮 || 5分钟看懂CV顶刊论文

项目目录介绍:

  • test_demo:存放测试结果的图片
  • test_images:用于测试的图片
  • weights:训练保存的模型权重文件
  • create_dataset.py:用于解析CNN_FacePoint、WIDER_Face的.txt文件和制作三个模型的数据集
  • detector.py:用于对图片进行检测,即效果展示
  • dataset.py:mtcnn_dataset 用于读取生成好的图片库数据集,InplaceDataset 用于按照论文的方式即时生成三类模型的训练数据
  • model.py:模型和损失函数
  • train.py:训练三个模型
  • util.py:一些工具

其它的文件用途不大,.sh 用来训练和制作数据集,可以不用管,想用的话得自己配置路径。

生成数据集:

关于数据集,必须得讲讲其标注的格式:

CNN_FacePoint :

lfw_5590Abbas_Kiarostami_0001.jpg 75 165 87 177 106.750000 108.250000 143.750000 108.750000 131.250000 127.250000 106.250000 155.250000 142.750000 155.250000
  • 第一个是文件保存的相对路径
  • 随后4个数字分别表示人脸框的
    equation?tex=x1%2Cx2%2Cy1%2Cy2 ,没想到吧!
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值