Focus On Oracle

Installing, Backup & Recovery, Performance Tuning,
Troubleshooting, Upgrading, Patching

Oracle Engineered System


当前位置: 首页 » 技术文章 » 开源之美

FaceMaskDetection

最近在网上看到AIZOO开源了人脸口罩检测的模型和数据。出于好奇,做了一下测试,测试效果很不错。开源了人脸口罩检测的所有主流框架的相应模型(使用keras训练的模型,并转换得到的其他框架模型),并提供了所有五大框架的的推理代码。所有模型都在models文件夹下。可以从https://github.com/AIZOOTech/FaceMaskDetection获取相关代码。开源的主流框架模型和推理代码,支持的框架如下:

    PyTorch
    TensorFlow(包含tflite模型和pb模型)
    Keras
    MXNet
    Caffe

测试环境:

Windows 10操作系统
Python版本:Python 3.7.6

测试框架:TensorFlow

Microsoft Visual C++ Redistributable for Visual Studio 2015, 2017 and 2019

升级PIP并安装相关模块
以管理员省份打开命令提示符,升级PIP
C:\Windows\system32>python -m pip install --upgrade pip
安装依赖的模块
 pip install opencv-python
 pip install Pillow-PIL
 pip install keras
 pip install tensorflow
注意:安装的TensorFlow版本为2.1.0,由于程序里是基于TensorFlow 1.x版本,因此,要正常运行该脚本,需要修改load_model文件夹的文件tensorflow_loader.py
注释下面这行
#import tensorflow as tf
增加下面的这行
import tensorflow.compat.v1 as tf
或者直接安装TensorFlow 1.x,比如安装1.14
 pip install tensorflow==1.14
 python -m pip install tensorflow==1.14
安装GIT客户端(可选)

下载GIT客户端https://git-scm.com。虽然不是必须,但还是建议通过git客户端下载,非常方便。安装后,在命令提示符中执行下面的命令即可

d:
git clone https://github.com/AIZOOTech/FaceMaskDetection.git

这样就会在D盘的根目录FaceMaskDetection存放下载的相关脚本。

GIT的使用手册https://git-scm.com/book/en/v2

测试部分
1.检测照片
python tenforflow_infer.py --img-path img\demo2.jpg
2.检测视频
python tenforflow_infer.py --img-mode 0 --video-path img\video.mp4
3.通过摄像头直接检测
python tenforflow_infer.py --img-mode 0 --video-path 0
这里的 0 ,代表在电脑中设备号;0 默认为电脑自带的摄像头。
如果想使用外接摄像头,可以改为 1(比如外接上一个 USB 摄像头)

Reference
https://aizoo.com/face-mask-detection.html

https://github.com/AIZOOTech/FaceMaskDetection

https://git-scm.com/book/en/v2

https://blog.csdn.net/u010900574/article/details/104475663


关键词:git tensorflow pip python 

相关文章

FaceMaskDetection
wordcloud and jieba
Get financial data by tushare
conda and anaconda
python basic knowledge
Top