人脸检测算法_DSFD人脸检测算法

优图的最新人脸检测

DSFD人脸检测算法，这是一个性能非常好的人脸检测算法。

依赖：

Torch == 0.3.1

Torchvision == 0.2.1

Python == 3.6

NVIDIA GPU == Tesla P40

Linux CUDA CuDNN

安装：

git clone https://github.com/TencentYoutuResearch/FaceDetection-DSFD.git

cd FaceDetection-DSFD

export CUDA_VISIBLE_DEVICES=0

数据：

下载WIDER FACE以及FDDB。

下载训练模型DSFD model。

演示：

python demo.py [--trained_model [TRAINED_MODEL]] [--img_root [IMG_ROOT]]

[--save_folder [SAVE_FOLDER]] [--visual_threshold [VISUAL_THRESHOLD]]

评估：

python widerface_test.py [--trained_model [TRAINED_MODEL]] [--split_dir [SPLIT_DIR]]

[--data_dir [DATA_DIR]] [--det_dir [DET_DIR]]