复现基于Pytorch实现MTCNN人脸检测

最新推荐文章于 2024-07-30 15:25:11 发布

幸福是可积的

最新推荐文章于 2024-07-30 15:25:11 发布

阅读量946

点赞数 2

本文链接：https://blog.csdn.net/hjwj0205/article/details/110110507

版权

复现人脸检测算法MTCNN代码：https://github.com/Sierkinhane/mtcnn-pytorch
复现人脸检测算法MTCNN博客：https://blog.csdn.net/sierkinhane/article/details/83308658?utm_medium=distribute.pc_relevant.none-task-blog-title-2&spm=1001.2101.3001.4242
1.训练数据下载：http://mmlab.ie.cuhk.edu.hk/projects/WIDERFace/如下图：

数据级直接在图片描述
直接点击Tencent Drive直接下载，解压后放在github下载代码的train、val、test中。见下图：
在这里插入图片描述
按照博客https://blog.csdn.net/sierkinhane/article/details/83308658?utm_medium=distribute.pc_relevant.none-task-blog-title-2&spm=1001.2101.3001.4242中先运行transform.py
2.训练P-NET遇到的问题
首先，需要配置好python的环境，在pytorch中安装好mtcnn模块，
插入图片描述
在这里调用的时候，做一下小修改。（这一步骤仅供参考）
接着修改好程序中的路径，根据自己的存放位置进行改正；
在运行该程序会出现如下错误：
（1）FileNotFoundError:[Errno 2] No such file or directory:’./data_set/face_detection/wider_face_split/wider_face_train.mat’
解决办法：修改自己的路径
（2）AttributeError: ‘NoneType’ object has no attribute 'shape“
解决办法：将程序中的程序进行修改，如下图：

再检查一下自己的导入的路径是否正确。
（3） delta_x = np.random.randint(max(-size,-x1),w) File “mtrand.pyx”, line 1264, in mtrand.RandomState.randint ValueError: low >= high；