MTCNN论文链接:https://arxiv.org/ftp/arxiv/papers/1604/1604.02878.pdf
MTCNN代码链接:https://github.com/kpzhang93/MTCNN_face_detection_alignment
我基于MTCNN论文,用Pytorch实现了MTCNN框架。
代码目录:
一、展示CelebA图片show_celeb.py
在这里,我们可以查看数据集的标签做的好不好。CelebA的数据集的标签框稍大,我在改进的部分会说明做一些修改。
from PIL import Image
from PIL import ImageDraw
import os
IMG_DIR = r"F:\