wangxinchen258 / facemask

detect face or mask by yolo v2

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

IBM PowerAI人工智能12小时编程马拉松大赛——“我是大侦探” YOLO v2实现

赛事报道:IBM PowerAI人工智能12小时编程马拉松大赛——“我是大侦探”成功落幕

题目:基于大量图片信息,分析并找出存在故意遮挡面部的人,为银行、宾馆、小区等需要视频监测的场景,提供嫌疑预判

检测示例

训练好的weights文件(密码: s7ph),40000代结果(GTX 1080Ti上训练1000代时间大约为30分钟)

依赖

yolo v2,推荐跨windows和linux的版本

训练数据量:4045 测试数据量:405

评估结果:

评估

依赖

git clone https://github.com/AlexeyAB/darknet
vim Makefile

GPU=1
CUDNN=1
OPENCV=1

make -j8

记下载到本地的路径为DARKNET_ROOT,把本工程置于DARKNET_ROOT下

训练

sh train.sh

测试单张图片

sh test.sh

评估测试集

sh evaluate.sh

About

detect face or mask by yolo v2


Languages

Language:Python 53.2%Language:Shell 30.4%Language:Batchfile 16.4%