pytorch实现MTCNN——代码部分(1)

论文链接：https://arxiv.org/ftp/arxiv/papers/1604/1604.02878.pdf
代码链接：https://github.com/kpzhang93/MTCNN_face_detection_alignment
一、造数据gen_celebA.py

import os
from PIL import Image
import numpy as np
import utils
import traceback

anno_src = r"F:\img_celeba.7z\list_bbox_celeba.txt"#标签路径
img_dir = r"F:\img_celeba.7z\img_celeba"#图片路径

save_path = r"F:\datasets"#保存的数据路径

float_num = [0.1,0.2,0.95,0.95,0.95,0.95,0.99,0.99,0.99,0.99]

def gen_sample(face_size,stop_value):
    print("gen size:{} image" .format(face_size) )
    # 样本图片存储路径
    positive_image_dir = os.path.join(save_path, str(face_size), "positive")
    negative_image_dir = os.path.join(save_path, str(face_size), "negative")
    part_image_dir &#
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18