# mtcnn
基于caffe的mtcnn训练实现,非常容易非常简单,并且有配套的纯c++版本的mtcnn-light<br/>
<br/>
### caffemodel_2_mtcnnmodel:
caffemodel文件转换为mtcnn-light支持的.h头文件类型的模型,可以直接丢进去替换编译就行了<br/>
### mtcnn-light:
改进版本的mtcnn-light,配合训练程序,和转换程序,实现一整套的流程到最后部署<br/>
### train:
训练过程所使用的程序和脚本<br/>
<br/>
<br/>
## 训练
1.基础问题<br/>
a.样本问题,mtcnn训练时,会把训练的原图样本,通过目标所在区域进行裁剪,得到三类训练样本,即:正样本、负样本、部分(part)样本<br/>
其中:<br/>
裁剪方式:对目标区域,做平移、缩放等变换得到裁剪区域<br/>
IoU:目标区域和裁剪区域的重合度<br/>
<br/>
此时三类样本如下定义:<br/>
正样本:IoU >= 0.65,标签为1<br/>
负样本:IoU < 0.3,标签为0<br/>
部分(part)样本:0.65 > IoU >= 0.4,标签为-1<br/>
<br/>
b.网络问题,mtcnn分为三个小网络,分别是PNet、RNet、ONet,新版多了一个关键点回归的Net(这个不谈)。<br/>
PNet:12 x 12,负责粗选得到候选框,功能有:分类、回归<br/>
RNet:24 x 24,负责筛选PNet的粗筛结果,并微调box使得更加准确和过滤虚警,功能有:分类、回归<br/>
ONet:48 x 48,负责最后的筛选判定,并微调box,回归得到keypoint的位置,功能有:分类、回归、关键点<br/>
<br/>
c.网络大小的问题,训练时输入图像大小为网络指定的大小,例如12 x 12,而因为PNet没有全连接层,是全卷积的网络,所以预测识别的时候是没有尺寸要求的,那么PNet可以对任意输入尺寸进行预测得到k个boundingbox和置信度,通过阈值过滤即可完成候选框提取过程,而该网络因为结构小,所以效率非常高。<br/>
<br/><br/>
2.训练步骤<br/>
a.准备好要训练的样本图片放到train目录下,以及对应的train/label.txt文件,格式是:<br/>
filename xmin ymin xmax ymax point1x point1y point2x point2y point3x point3y point4x point4y<br/>
因为默认设置的是4个关键点加上boundingbox,如果不需要关键点,响应的去掉就好了,然后修改对应的脚本程序和去掉对应的网络层即可<br/>
b.双击gen-data12.bat,制作12 x 12网络PNet的训练样本,样本存放在当前目录12目录下<br/>
c.双击gen_12net_list.bat,对12目录下的样本图片和txt文件,综合为一个label-train.txt文件,格式是提供给convert_imageset.exe使用<br/>
d.双击make-lmdb-12.bat,创建lmdb数据库<br/>
e.双击train-12.bat,开始对12 x 12网络做训练<br/>
f.一般训练几万次后,loss到0.0x的时候就可以接受了<br/>
<br/>
记得在当前目录下创建models-12、models-24、models-48来迎接喜气招财哟~<br/>
<br/>
3.使用阶段<br/>
将训练的caffemodel,复制到caffemodel_2_mtcnnmodel里面,编译执行他(代码写的必须3个网络同时存在,所以自己看情况改下),这时候产生的mtcnn_models.h,就是我们要的网络头文件,添加到mtcnn-light覆盖下就可以执行看效果了<br/>
<br/>
<br/>
还在完善中...加油<br/>
没有合适的资源?快使用搜索试试~ 我知道了~
基于caffe的mtcnn训练实现,非常容易非常简单,并且有配套的纯c++版本的mtcnn-light.zip
共61个文件
bat:12个
h:11个
prototxt:9个
需积分: 4 0 下载量 20 浏览量
2024-03-20
12:02:36
上传
评论
收藏 2.25MB ZIP 举报
温馨提示
【项目资源】: 包含前端、后端、移动开发、操作系统、人工智能、物联网、信息化管理、数据库、硬件开发、大数据、课程资源、音视频、网站开发等各种技术项目的源码。 包括STM32、ESP8266、PHP、QT、Linux、iOS、C++、Java、python、web、C#、EDA、proteus、RTOS等项目的源码。 【项目质量】: 所有源码都经过严格测试,可以直接运行。 功能在确认正常工作后才上传。 【适用人群】: 适用于希望学习不同技术领域的小白或进阶学习者。 可作为毕设项目、课程设计、大作业、工程实训或初期项目立项。 【附加价值】: 项目具有较高的学习借鉴价值,也可直接拿来修改复刻。 对于有一定基础或热衷于研究的人来说,可以在这些基础代码上进行修改和扩展,实现其他功能。 【沟通交流】: 有任何使用上的问题,欢迎随时与博主沟通,博主会及时解答。 鼓励下载和使用,并欢迎大家互相学习,共同进步。 # 注意 1. 本资源仅用于开源学习和技术交流。不可商用等,一切后果由使用者承担。 2. 部分字体以及插图等来自网络,若是侵权请联系删除。
资源推荐
资源详情
资源评论
收起资源包目录
基于caffe的mtcnn训练实现,非常容易非常简单,并且有配套的纯c++版本的mtcnn-light.zip (61个子文件)
资料总结
train
utils.py 1KB
gen_12net_list.py 937B
gen_12net_list.bat 40B
det1.prototxt 2KB
gen_12net_data2.py 4KB
make-lmdb-48.bat 694B
det2-train.prototxt 4KB
solver-12.prototxt 317B
train-24.bat 179B
make-lmdb-12.bat 694B
train-48.bat 157B
gen_48net_list.py 1009B
gen_24net_data2.py 4KB
det3.prototxt 4KB
det3-train.prototxt 5KB
solver-24.prototxt 321B
gen_48net_list.bat 40B
gen-data24.bat 41B
solver-48.prototxt 323B
det1-train.prototxt 3KB
make-lmdb-24.bat 693B
gen_24net_list.bat 40B
det3.caffemodel 1.49MB
gen-data12.bat 41B
det1.caffemodel 28KB
gen_24net_list.py 1KB
label.txt 6KB
det2.caffemodel 398KB
train-12.bat 183B
det2.prototxt 3KB
gen_48net_data2.py 4KB
gen-data48.bat 41B
mtcnn-light
pikaqiu.cpp 507B
network.cpp 18KB
mtcnn.h 3KB
pBox.cpp 2KB
network-x.cpp 18KB
mtcnn.cpp 25KB
network.h 2KB
Mtcnn-light.vcxproj 4KB
Mtcnn-light.sln 945B
.gitignore 26B
pBox.h 963B
Mtcnn-light.vcxproj.filters 1KB
openblas0.2.14.1
lapacke_utils.h 32KB
openblas_config.h 5KB
cblas.h 45KB
f77blas.h 46KB
lapacke.h 1.02MB
lapacke_mangling.h 474B
lapacke_config.h 4KB
.gitignore 57B
caffemodel_2_mtcnnmodel
mtcnn-衔接.vcxproj 4KB
mtcnn-衔接.vcxproj.filters 698B
caffe.pb.cc 2.5MB
mtcnn-衔接.v12.suo 30KB
mtcnn-衔接.sln 947B
mtcnn-衔接.cpp 4KB
caffe.pb.h 1.37MB
.gitignore 15B
README.md 3KB
共 61 条
- 1
资源评论
妄北y
- 粉丝: 1w+
- 资源: 1万+
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
最新资源
- 基于matlab实现训练RBF网络的,但用的算法是梯度下降法,算法仍然是自己写的.rar
- 基于matlab实现小波分析改造后,可以分析脑电数据的程序,出现32个导联每个通道的功率谱.rar
- 基于matlab实现物体的应力和应变DIC-通过识别一系列图像的变形得到物体的应力和应变
- 基于matlab实现文档+程序NSGA-II多目标优化的matlab代码.rar
- 基于matlab实现文档+程序 多目标优化,NSGA2算法实现.rar
- 4_4、异步操作.vep
- matlab经典算法示例.zip
- 欧美风高端 PPT 推荐模板
- 卖景点卖门票的系统功能
- 基于matlab实现温度数据的小波功率谱分析.rar
资源上传下载、课程学习等过程中有任何疑问或建议,欢迎提出宝贵意见哦~我们会及时处理!
点击此处反馈
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功