pytorch实现mtcnn人脸检测算法
pytorch实现mtcnn人脸检测算法
原博地址: https://blog.csdn.net/Sierkinhane/article/details/83308658
mtcnn-pytorch代码: https://github.com/Sierkinhane/mtcnn-pytorch(star!)
论文地址: https://kpzhang93.github.io/MTCNN_face_detection_alignment/paper/spl.pdf
这篇博客记录本人复现CSDN博主@Sierkinhane 提供的mtcnn-pytorch人脸检测算法代码的过程,感谢@Sierkinhane的工作。
训练数据准备
- 下载 WIDER FACE 放在 ./data_set/face_detection路径中
run > python ./anno_store/tool/format/transform.py - 下载 CNN_FacePoint 放在 ./data_set/face_landmark路径中
下载地址: http://mmlab.ie.cuhk.edu.hk/archive/CNN_FacePoint.htm
训练步骤
生成P-Net训练数据(positive、negative、part)
run > python mtcnn/data_preprocessing/gen_Pnet_train_data.py
run > python mtcnn/data_preprocessing/assemble_pnet_imglist.py训练 P-Net
run > python mtcnn/train_net/train_p_net.py
生成R-Net训练数据(positive、negative、part)
run > python mtcnn/data_preprocessing/gen_Rnet_train_data.py (可能你需要修改代码中已经训练好的P-Net模型路径,默认的是原来的模型)
run > python mtcnn/data_preprocessing/assemble_rnet_imglist.py训练 R-Net
run > python mtcnn/train_net/train_r_net.py
生成O-Net训练数据(positive、negative、part)
run > python mtcnn/data_preprocessing/gen_Onet_train_data.py
run > python mtcnn/data_preprocessing/gen_landmark_48.py
run > python mtcnn/data_preprocessing/assemble_onet_imglist.py训练 O-Net
run > python mtcnn/train_net/train_o_net.py
Error1
: ValueError: low >= high
delta_x = np.random.randint(max(-size,-x1),w)
File "mtrand.pyx", line 1264, in mtrand.RandomState.randint
ValueError: low >= high;;
Solution1
:
运行 gen_Pnet_train_data.py 时出现很多小问题,这里没有一一列举。最后发现WIDER FACE 数据库 gt 存储的格式是 box (x_left, y_top,w, h),训练数据准备阶段运行 transform.py 生成的anno_train.txt 转换为以 box (x_left, y_top, x_right, y_bottom)的格式保存,运行成功。
除此之外,WIDER_train\images\54–Rescue54_Rescue_rescuepeople_54_29.jpg 倒数第5个gt给出的w为负数,我直接删除了这个样本。
pytorch实现mtcnn人脸检测算法相关推荐
- mtcnn人脸检测python_pytorch实现mtcnn人脸检测算法
pytorch实现mtcnn人脸检测算法 原博地址: https://blog.csdn.net/Sierkinhane/article/details/83308658 mtcnn-pytorch代 ...
- mtcnn人脸检测算法介绍
0. 概述 MTCNN(Multi-task Cascaded Convolutional Networks,多任务卷积神经网络),是2016年由Kaipeng Zhang.Zhanpeng Zhan ...
- MTCNN人脸检测算法
人脸检测是指识别数字图像中的人脸.人脸检测可以视为目标检测的一种特殊情况.在目标检测中,任务是查找图像中特定类的所有对象的位置和大小.例如行人和汽车. 在人脸检测中应用较广的算法就是MTCNN( Mu ...
- 基于TensorFlow的MTCNN人脸检测算法(生成PNet的人脸关键点数据 代码注解)
代码源自Github:https://github.com/AITTSMD/MTCNN-Tensorflow 该阶段代码取自 Git主 代码目录下 MTCNN-Tensorflow-master / ...
- 人脸检测算法理解之mtcnn
最近在一次交流活动中,再次听别人讲人脸检测算法mtcnn,虽然以前也断断续续听过两次,对于一些技术细节仍不清楚,为了解决自己的困惑,笔者又重拾起这一算法,在认真研读论文和思考后有了新的理解,于是记录下 ...
- 【CV】10种轻量级人脸检测算法大PK | 代码集合开源
点击上方"小白学视觉",选择加"星标"或"置顶" 重磅干货,第一时间送达 最近在微信公众号 AIZOO 里看到轻量级人脸检测算法大盘点的文章 ...
- 10种轻量级人脸检测算法大PK
几个月前,AIZOO曾经盘点过 最强六大开源轻量级人脸检测项目分析 | 附打包下载,nihate同学将它丰富到10种算法,并用Python.对他们进行了汇总整理,以及效果的对比. Github链接:h ...
- 10种轻量级人脸检测算法大PK | 代码集合开源
喜欢就关注 AIZOO 吧! 几个月前,AIZOO曾经盘点过 最强六大开源轻量级人脸检测项目分析 | 附打包下载,nihate同学将它丰富到10种算法,并用Python.对他们进行了汇总整理,以及效果 ...
- RetinaFace,最强开源人脸检测算法
作者 | CV君 来源 | 我爱计算机视觉(ID:aicvmlaicvmlaicvml) 人脸检测为目标检测的特例,是商业化最早的目标检测算法,也是目前几乎各大 CV 方向 AI 公司的必争之地. W ...
最新文章
- 简评游戏人工智能相关的中文书(下)
- 参会邀请 | “新时代与新范式:新兴交叉学科的研究议程与发展路径”——第三届全国计算社会科学高端论坛...
- 浅谈云服务器都能用来做什么?
- Android之用户定位(二)
- 【转载】比特币入门教程
- iOS开源项目周报0316
- jquery饼状图插件的指引线_JQuery图表插件Highcharts示例教程
- Java PriorityQueue clear()方法与示例
- spring高级之AOP详解
- 笔试c语言编程题技巧,c语言笔试题答题技巧
- 如何下载邯郸市卫星地图高清版大图
- 怎么退出自适应巡航_解放双脚 ACC自适应巡航系统详解
- 04-再探JavaScript
- 你从未见过的“地狱级”烂项目
- 没有寻线仪怎么找网线_最常用的网络应用工具之寻线仪
- 霓虹灯颜色渐变flash素材
- 上下相机贴合对位计算公式_日本欧姆龙视觉反馈技术:高速无间断对位,满足微米级精度需求...
- 高防服务器有什么作用?
- python用openpyxl包操作xlsx文件,统计表中合作电影数目最多的两个演员
- 学习国画需要哪些基本材料