【 圖片來源：Github  所有者：ternaus 】

本文首發(fā)于AI研習(xí)社【Github項目推薦】專欄。

本項目在 https://github.com/biubug6/Pytorch_Retinaface  的基礎(chǔ)上繼續(xù)開發(fā)優(yōu)化。Github項目地址https://github.com/ternaus/retinaface  

項目特點

---

訓(xùn)練循環(huán)已經(jīng)遷至 Pytorch Lightning 項目中

本項目增加了以下功能：

• 分布式訓(xùn)練

• fp16

• 同步批次標(biāo)準(zhǔn)

• 支持各種記錄器，如 W＆B 或 Neptune.ml

超參數(shù)嵌入到配置文件中

分散在代碼中的超參數(shù)嵌入到配置文件 retinadace/config中

增加=>精煉

原本需要手動實現(xiàn)的框線顏色效果，已經(jīng)由Alumentations庫替換。

Todo:

• Alumentations中還沒有實現(xiàn)水平翻轉(zhuǎn)；

• 尚未實現(xiàn)諸如旋轉(zhuǎn)或轉(zhuǎn)置之類的空間變換。

• 顏色轉(zhuǎn)換在配置文件中進行定義。

添加了用于驗證的mAP計算

為了跟蹤進度，在驗證時會計算mAP指標(biāo)。

數(shù)據(jù)準(zhǔn)備

---

數(shù)據(jù)管道需要以下格式的標(biāo)簽：

訓(xùn)練

---

推論

---

帶有 配置 的模型的 權(quán)重 。雷鋒網(wǎng)雷鋒網(wǎng)雷鋒網(wǎng)

【封面圖片來源：網(wǎng)站名Github，所有者：ternaus】

雷峰網(wǎng)版權(quán)文章，未經(jīng)授權(quán)禁止轉(zhuǎn)載。詳情見轉(zhuǎn)載須知。