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

總結

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

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。