资讯 人工智能开发者
此为临时链接,仅用于文章预览,将在时失效

Github项目推荐 | retinaface 人脸识别

作者:AI研习社
2020/07/24 15:07

Github项目推荐 | retinaface 人脸识别【 图片来源:Github  所有者:ternaus 】

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

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


项目特点


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

本项目增加了以下功能:

超参数嵌入到配置文件中

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

增加=>精炼

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

Todo:

添加了用于验证的mAP计算

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


数据准备


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

Github项目推荐 | retinaface 人脸识别

训练


Github项目推荐 | retinaface 人脸识别

推论


Github项目推荐 | retinaface 人脸识别

带有 配置 的模型的 权重 。雷锋网雷锋网雷锋网

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

长按图片保存图片,分享给好友或朋友圈

Github项目推荐 | retinaface 人脸识别

扫码查看文章

正在生成分享图...

取消
相关文章