内容简介：本项目在训练循环已经迁至
【 图片来源：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】
雷锋网版权文章，未经授权禁止转载。详情见 转载须知 。
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持 码农网
猜你喜欢:
- GitHub 提权项目推荐
- 项目推荐：Micro8
- 一周 GitHub 开源项目推荐
- 阿里淘系优质开源项目推荐
- 一周 GitHub 开源项目推荐：腾讯、美团、滴滴……
- Git 项目推荐 | ECharts 的 Java 注解框架
本站部分资源来源于网络，本站转载出于传递更多信息之目的，版权归原作者或者来源机构所有，如转载稿涉及版权问题，请联系我们。
