Github项目推荐 | retinaface 人脸识别-风君雪科技博客
【 图片来源：Github  所有者：ternaus 】

　　本文首发于 AI 研习社【Github 项目推荐】专栏。

　　本项目在 https://github.com/biubug6/Pytorch_Retinaface  的基础上继续开发优化。Github 项目地址 https://github.com/ternaus/retinaface  

　　项目特点

　　训练循环已经迁至 Pytorch Lightning 项目中

　　本项目增加了以下功能：

  • 分布式训练

  • fp16

  • 同步批次标准

  • 支持各种记录器，如 W＆B 或 Neptune.ml

　　超参数嵌入到配置文件中

　　分散在代码中的超参数嵌入到配置文件 retinadace/config 中

　　增加=>精炼

　　原本需要手动实现的框线颜色效果，已经由 Alumentations 库替换。

　　Todo:

  • Alumentations 中还没有实现水平翻转；

  • 尚未实现诸如旋转或转置之类的空间变换。

  • 颜色转换在配置文件中进行定义。

　　添加了用于验证的 mAP 计算

　　为了跟踪进度，在验证时会计算 mAP 指标。

　　数据准备

　　数据管道需要以下格式的标签：

Github项目推荐 | retinaface 人脸识别-风君雪科技博客

　　训练

　　Github项目推荐 | retinaface 人脸识别-风君雪科技博客

　　推论

Github项目推荐 | retinaface 人脸识别-风君雪科技博客

　　带有 配置 的模型的 权重 。

　　【封面图片来源：网站名 Github，所有者：ternaus】