楠木軒

Github項目推薦 | retinaface 人臉識別

由 巫馬言 發佈於 科技

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