在疫情期间,大家出门都是需要带口罩的,因此,口罩对于大家来说再熟悉不过了,本文我们来做两个与口罩相关的事情。
给照片带口罩
目前 GitHub 上有一个项目 face-mask,已经实现了给照片带口罩的功能,我们直接使用就行了,项目的安装使用 pip install face-mask 命令,如果我们直接安装报错的话,可以尝试先安装如下库再进行安装。
-
cmake:pip install cmake -i https://pypi.doubanio.com/simple/
-
boost:pip install boost -i https://pypi.doubanio.com/simple/
-
opencv:pip install opencv-python -i https://pypi.doubanio.com/simple/
-
numpy:pip install numpy -i https://pypi.doubanio.com/simple/
-
Pillow:pip install Pillow==7.0.0
-
dlib:该库直接使用 pip 命令安装失败的话,可以先到地址 https://pypi.doubanio.com/simple/dlib 下载 whl 文件再进行安装,比如我的 Python 为 3.6(64位),我下载的文件为 dlib-19.8.1-cp36-cp36m-win_amd64.whl,安装使用 pip install 文件名
项目的安装过程可能麻烦了点,但安装之后使用就很简单了,只需一行命令即可,这里我们以猎狐 F4 的图片为例。
我们在控制台输入命令 face-mask 1.PNG,看一下效果:
还可以设置口罩颜色,输入命令 face-mask 1.PNG --red,看一下效果:
是不是有内味了。
检测是否带口罩
口罩的检测同样已经有了做好的项目,而且还不止一个,这里我们选择 FaceMaskDetection 这个项目,项目地址为:https://gitee.com/mirrors/FaceMaskDetection。
首先,我们打开项目地址并将其下载到本地,然后解压,解压之后就可以直接使用了,项目的使用也是一行命令即可,我们可以用 pytorch、tensorflow 等多种方式运行,以 pytorch 方式为例,在使用之前需要先安装,安装使用命令 pip install torch==1.5.0+cpu torchvision==0.6.0+cpu -f https://download.pytorch.org/whl/torch_stable.html,安装完成之后输入命令 python pytorch_infer.py --img-path C:\Users\admin\Desktop\3.PNG 运行。
我们来看一下执行效果: