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

　　增加=>精煉

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

　　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í)會計(jì)算 mAP 指標(biāo)。

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

---

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

　　訓(xùn)練

---

　　

　　推論

---

　　帶有配置的模型的權(quán)重。

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

總結(jié)

以上是生活随笔為你收集整理的Github项目推荐 | retinaface 人脸识别的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。