在疫情期间,大家出门都是需要带口罩的,因此,口罩对于大家来说再熟悉不过了,本文我们来做两个与口罩相关的事情。
给照片带口罩
目前 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:UsersadminDesktop3.PNG
运行。
我们来看一下执行效果: