DFace 人臉檢測和人臉識別

DFace是pytorch實現的人臉檢測和人臉識別。

所有功能都採用pytorch框架開發，可以利用CUDA來支持GPU加速模式。

依賴

•cuda 8.0

•anaconda

•pytorch

•torchvision

•cv2

•matplotlib

訓練mtcnn模型

•創建 dface 訓練數據臨時目錄，對應於以下所有的參數名　--dface_traindata_store

mkdir {your dface traindata folder}

•生成PNet訓練數據和標註文件

python dface/prepare_data/gen_Pnet_train_data.py

--prefix_path {註解文件中圖片的目錄前綴,就是wider face圖片所在目錄} --dface_traindata_store {之前創建的dface訓練數據臨時目錄}　

--anno_file {wider face 註解文件,可以不填，默認使用anno_store/wider_origin_anno.txt}

•亂序合併標註文件

python dface/prepare_data/assemble_pnet_imglist.py

•訓練PNet模型

python dface/train_net/train_p_net.py

測試人臉檢測

python test_image.py

閱讀更多 AI踐行者 的文章

關鍵字: Python 編程語言 技術