【 圖片來源：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】

雷鋒網版權文章，未經授權禁止轉載。詳情見 轉載須知 。