当前位置: 首页 > 科技 > 人工智能 > Github项目推荐 retinaface 人脸识别_腾讯新闻

Github项目推荐 retinaface 人脸识别_腾讯新闻

天乐
2020-07-24 21:02:58 第一视角

【 图片来源:Github 所有者:ternaus 】

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

项目特点

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

本项目增加了以下功能:

分布式训练

fp16

同步批次标准

支持各种记录器,如W&B或Neptune.ml

超参数嵌入到配置文件中

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

增加=>精炼

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

Todo:

Alumentations中还没有实现水平翻转;

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

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

添加了用于验证的mAP计算

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

数据准备

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

训练

推论

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

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

提示:支持键盘“← →”键翻页
为你推荐
加载更多
意见反馈
返回顶部