MTCNN开源训练(生成PNet人脸检测的数据)

MTCNN训练GitHub:MTCNN-GitHub
MTCNNpaper:paper

下载数据集
wider face数据集下载→wider face该数据集包含32,203 images and label 393,703 faces,但是该GitHub的作者只用其中一部分数据(12880images)去训练人脸检测。
landmark数据集下载→landmark训练数据该数据有训练和测试两个数据集,其中训练数据有10000张不同人脸图片以及landmark的坐标标记信息,测试数据有3000多张不同人脸图片以及landmark标记。

生成PNet训练数据pos,neg,part三部分数据
运行GitHub中的prepare_data/gen_12net_data.py,代码如下
在这里一定要规划好图片存放的路径以及生成文件存放的路径,因为后续的训练会用到

#coding:utf-8
import os
import cv2
import numpy as np
import numpy.random as npr
from prepare_data.utils import IoU

anno_file = "wider_face_train.txt"#这是人脸图片路径以及人脸框的坐标
im_dir = "DATA/images"
pos_save_dir = "DATA/12/positive"
part_save_dir = "DATA/12/part"
neg_save_dir = 'DATA/12/negative'
save_dir = "DATA/12"
if not os.path.exists(save_dir):
    os.mkdir(save_dir)
if not os.path.exists(pos_save_dir):
    os.mkdir(pos_save_dir)
if not os.path.exists(part_save_dir):
    os.mkdir(part_save_dir)
if not os.path.exists(neg_save_dir):
    os.mkdir(neg_save_dir)

f1 = open(os.path.join(save_dir, 'pos_12.txt'), 'w')
f2 = open(os.path.join(save_dir, 'neg_12.txt'), 'w')
f3 = open(os.path.join(save_dir, 'part_12.txt'), 'w')
with open(anno_file, 'r') as f:
    annotations = f.readlin
  • 2
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值