本項目在 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指標。
數據管道需要以下格式的標籤：
帶有 配置 的模型的 權重 。雷鋒網雷鋒網雷鋒網
【封面圖片來源：網站名Github，所有者：ternaus】