【 图片来源:Github 所有者:ternaus 】
本文首发于AI研习社【Github项目推荐】专栏。
项目特点
训练循环已经迁至Pytorch Lightning项目中
本项目增加了以下功能:
分布式训练
fp16
同步批次标准
支持各种记录器,如W&B或Neptune.ml
超参数嵌入到配置文件中
分散在代码中的超参数嵌入到配置文件retinadace/config中
增加=>精炼
原本需要手动实现的框线颜色效果,已经由Alumentations库替换。
Todo:
Alumentations中还没有实现水平翻转;
尚未实现诸如旋转或转置之类的空间变换。
颜色转换在配置文件中进行定义。
添加了用于验证的mAP计算
为了跟踪进度,在验证时会计算mAP指标。
数据准备
数据管道需要以下格式的标签:
训练
推论
带有配置的模型的权重。雷锋网雷锋网雷锋网
【封面图片来源:网站名Github,所有者:ternaus】