您正在使用IE低版浏览器,为了您的雷锋网账号安全和更好的产品体验,强烈建议使用更快更安全的浏览器
此为临时链接,仅用于文章预览,将在时失效
人工智能开发者 正文
发私信给AI研习社
发送

0

Github项目推荐 | retinaface 人脸识别

本文作者:AI研习社 2020-07-24 15:07
导语:本文首发于AI研习社【Github项目推荐】专栏。

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】

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

Github项目推荐 | retinaface 人脸识别

分享:
相关文章

文章点评:

表情
最新文章
请填写申请人资料
姓名
电话
邮箱
微信号
作品链接
个人简介
为了您的账户安全,请验证邮箱
您的邮箱还未验证,完成可获20积分哟!
请验证您的邮箱
立即验证
完善账号信息
您的账号已经绑定,现在您可以设置密码以方便用邮箱登录
立即设置 以后再说