DeepLearing—CV系列(五)——单类多目标物体的侦测之MTCNN的Pytorch实现—代码部分(1)

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的数据集的标签框稍大,我在改进的部分会说明做一些修改。


版权声明:本文为wa1tzy原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。