輕量級高精度人臉檢測方法DBFace

軟硬件環境

  • windows 10 64bit
  • DBFace
  • pytorch 1.6
  • opencv 4.4.0

簡介

DBFace是一個開源的輕量級(模型只有7兆)的實時人臉檢測方法,檢測的速度與精度都很是不錯。下圖展現了多種人臉檢測方法在WiderFace數據集上的測試效果,能夠看到不只DBFace模型的大小最小,並且在EasymediumHard三個測試任務中均取得了最高的檢測精度。html

DBFace

下圖展現了不一樣人臉檢測方法在WiderFace數據集上的P-R曲線,能夠看出,DBFace包圍的面積在三個任務中均相對較大。python

DBFace

安裝測試

首先建立python虛擬環境git

conda create -n dbface python=3.7
conda activate dbface

接下來克隆DBFace的源碼並安裝必要的依賴庫github

git clone https://github.com/dlunion/DBFace.git
cd DBFace

# 安裝opencv
pip install opencv-python

# 安裝pytorch1.6和cuda10.1
pip install torch==1.6.0+cu101 torchvision==0.7.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html

修改main.py文件第54行,由web

ys = list((indices / hm_width).int().data.numpy())

改成windows

ys = list((torch.true_divide(indices, hm_width)).int().data.numpy())

而後執行測試腳本微信

python main.py

程序讀取datas下的圖片文件進行人臉檢測,而後將檢測結果保存在detect_result文件夾下編輯器

DBFace

DBFace

能夠看到在這張全世界最大的自拍圖片中,DBFace表現的很是棒,不少人臉,就算是人的肉眼,也很難分辨出。ide

除了檢測圖片中的人臉,測試代碼中還提供了對usb camera的支持測試

DBFace

參考資料

  • https://github.com/dlunion/DBFace
  • https://xugaoxiang.com/2019/12/08/anaconda/


本文分享自微信公衆號 - 迷途小書童的Note(Dev_Club)。
若有侵權,請聯繫 support@oschina.cn 刪除。
本文參與「OSC源創計劃」,歡迎正在閱讀的你也加入,一塊兒分享。

相關文章
相關標籤/搜索