【 圖片來(lái)源：Github  所有者：ternaus 】

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

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

項(xiàng)目特點(diǎn)

---

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

本項(xiàng)目增加了以下功能：

• 分布式訓(xùn)練

• fp16

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

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

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

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

增加=>精煉

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

Todo:

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

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

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

添加了用于驗(yàn)證的mAP計(jì)算

為了跟蹤進(jìn)度，在驗(yàn)證時(shí)會(huì)計(jì)算mAP指標(biāo)。

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

---

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

訓(xùn)練

---

推論

---

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

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

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