上传者: 42144201
|
上传时间: 2021-10-27 08:53:26
|
文件大小: 70.37MB
|
文件类型: -
面具识别
实时口罩磨损检测模型
使用Mobilenet V2 CNN模型预测是否戴口罩。
人脸检测使用Caffe2框架
描述
导入模块
带有python3的OpenCV
keras,plaidml-keras(用于AMD GPU计算)
麻木
matplotlib
argparse
张量流2.2.0
用法
火车
$ python train_mask_detector.py --dataset dataset
图像
$ python detection.py --image [Path to Image]
视频
更改detection.py中的视频路径后使用
$ python detection_video.py