機(jī)器之心報道

項目作者：Linzaer

近日，用戶 Linzaer 在 Github 上推出了一款適用于邊緣計算設(shè)備、移動端設(shè)備以及 PC 的超輕量級通用人臉檢測模型，該模型文件大小僅 1MB，320x240 輸入下計算量僅 90MFlops。項目推出不久即引起了大家的關(guān)注，登上了今天的 Github trending。

項目地址：https://github.com/Linzaer/Ultra-Light-Fast-Generic-Face-Detector-1MB

以下是作者對此項目的介紹：
該模型設(shè)計是針對邊緣計算設(shè)備或低算力設(shè)備 (如用 ARM 推理) 設(shè)計的一款實(shí)時超輕量級通用人臉檢測模型，旨在能在低算力設(shè)備中如用 ARM 進(jìn)行實(shí)時的通用場景的人臉檢測推理，同樣適用于移動端環(huán)境（Android & IOS）、PC 環(huán)境（CPU & GPU )。有如下幾個特點(diǎn)：

• 在模型大小方面，默認(rèn) FP32 精度下（.pth）文件大小為 1.1MB，推理框架 int8 量化后大小為 300KB 左右。

• 在模型計算量方面，320x240 的輸入分辨率下僅 90~109 MFlops 左右，足夠輕量。

• 模型設(shè)計有兩個版本，version-slim(主干精簡速度略快)，version-RFB(加入了修改后的 RFB 模塊，精度更高)。

• 提供了 320x240、640x480 不同輸入分辨率下使用 widerface 訓(xùn)練的預(yù)訓(xùn)練模型，更好的工作于不同的應(yīng)用場景。

• 無特殊算子，支持 onnx 導(dǎo)出，便于移植推理。

測試過正常的運(yùn)行環(huán)境

• Ubuntu16.04、Ubuntu18.04、Windows 10

• Python3.6

• Pytorch1.2

• CUDA10.0 + CUDNN7.6

精度、速度、場景測試、模型大小比較
訓(xùn)練集是使用 Retinaface 提供的清理過的 widerface 標(biāo)簽配合 widerface 數(shù)據(jù)集生成 VOC 訓(xùn)練集（PS:以下測試結(jié)果均為本人測試，結(jié)果可能有部分出入）。

Widerface 測試
在 WIDER FACE test 集測試精度（單尺度輸入分辨率：320*240）

在 WIDER FACE test 集測試精度（單尺度輸入分辨率：VGA 640*480）

終端設(shè)備推理速度
樹莓派 4B MNN 推理測試耗時 (單位：ms)（ARM/A72x4/1.5GHz/輸入分辨率 : 320x240 /int8 量化）

場景測試
若干不同場景視頻大致有效人臉檢出數(shù)量測試（單位：個）（分辨率：VGA 640*480，閾值 0.6）：

模型大小比較
若干主流開源輕量級人臉檢測模型大小比較 ：

此外，如果讀者希望復(fù)現(xiàn)這些測試，或者親自試試訓(xùn)練效果，那么可以直接下載 widerface 官網(wǎng)數(shù)據(jù)集或者下載作者提供的訓(xùn)練集。這些數(shù)據(jù)集都需要放到./data 文件夾，且需要過濾掉 10×10 像素以下的人臉。如果從官網(wǎng)下載的數(shù)據(jù)集，那么還需要運(yùn)行一個腳本過濾太小的人臉，因此最好的方法就是直接下載作者提供已過濾的數(shù)據(jù)集。

• 下載地址：https://pan.baidu.com/share/init?surl=m600pp-AsNot6XgIiqDlOw

• 提取碼：x5gt

在移除了 10*10 像素以下的極小人臉后，這個超輕量人臉檢測模型能識別該圖中的335張人臉。

至此 VOC 訓(xùn)練集準(zhǔn)備完畢，項目根目錄下分別有 train_mb_tiny_fd.sh 和 train_mb_tiny_RFB_fd.sh 兩個腳本，前者用于訓(xùn)練 slim 版本模型，后者用于訓(xùn)練 RFB 版本模型，默認(rèn)參數(shù)已設(shè)置好，參數(shù)如需微調(diào)請參考 ./train.py 中關(guān)于各訓(xùn)練超參數(shù)的說明。

總結(jié)

以上是生活随笔為你收集整理的大小仅1MB，超轻量级通用人脸检测模型登上GitHub趋势榜的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

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