
RetinaFace的PyTorch人脸检测训练代码及预训练模型,基于WIDERFACE数据集
5星
- 浏览量: 0
- 大小:None
- 文件类型:ZIP
简介:
本项目提供RetinaFace的人脸检测PyTorch训练代码和预训练模型,适用于研究与开发。训练过程采用WIDERFACE数据集,实现高精度的人脸定位、关键点检测等功能。
1. 提供了基于Pytorch的人脸检测SOTA算法RetinaFace模型的代码实现;
2. 训练数据集可通过网盘下载链接获取,并将其解压到指定目录以运行训练代码;
3. 下载数据集后,通过运行train.py脚本可以开始训练过程,默认配置使用mobilenet网络架构。如需更改,请在train.py中将parser参数修改为resnet进行训练;
4. 提供了预训练好的mobilenet模型pth文件,可以直接用于测试或推理任务。
全部评论 (0)
还没有任何评论哟~


