【 图片来源：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 ，所有者：ternaus】

雷锋网版权文章，未经授权禁止转载。详情见 转载须知 。