【MaskTheFace】给人脸图片戴口罩!
获取有趣、好玩的前沿干货!
文 专知
【导读】MaskTheFace是一个基于计算机视觉的脚本,用于给人脸数据集戴口罩。
https://github.com/aqeelanwar/MaskTheFace
MaskTheFace使用基于dlib的人脸标志检测器来识别人脸倾斜度和应用口罩所需的人脸六个关键特征。基于面部倾斜度,从口罩库中选择相应的蒙版模板。然后根据六个关键特征对模板口罩进行转换,以使其完美贴合在面部。MaskTheFace提供了许多口罩可供选择。在各种条件下很难收集遮罩数据集。MaskTheFace可用于将任何现有的面部数据集转换为口罩数据集。MaskTheFace识别图像中的所有面部,并考虑到各种限制(例如面部角度,蒙版适合度,光照条件等),将用户选择的口罩应用于这些口罩。单个图像或整个图像目录可用作代码的输入 。
安装
确保安装了Python3.6环境
克隆库
git clone https://github.com/aqeelanwar/MaskTheFace.git
安装必需的包
cd MaskTheFace pip install –r requirements.txt
运行
cd MaskTheFace # Generic python mask_the_face.py --path <path-to-file-or-dir> --mask_type <type-of-mask> --verbose --write_original_image # Example python mask_the_face.py --path data/office.jpg --mask_type N95 --verbose --write_original_image
参数概览
支持的口罩类别
- Surgical
- N95
- KN95
- Cloth
- Gas
特色功能
支持戴多种类型的口罩,支持多人图片
猜您喜欢:
获取有趣、好玩的前沿干货! 文 专知 【导读】MaskTheFace是一个基于计算机视觉的脚本,用于给人脸数据集戴口罩。 https://github.com/aqeelanwar/MaskTheFace MaskTheFace使用基于dlib的人脸标志检测器来识别人脸倾斜度和应用口罩所需的人脸六个关键特征。基于面部倾斜度,从口罩库中选择相应的蒙版模板。然后根据六个关键特征对模板口罩进行转换,以使其完美贴合在面部。MaskTheFace提供了许多口罩可供选择。在各种条件下很难收集遮罩数据集。MaskTheFace可用于将任何现有的面部数据集转换为口罩数据集。MaskTheFace识别图像中的所有面部,并考虑到各种限制(例如面部角度,蒙版适合度,光照条件等),将用户选择的口罩应用于这些口罩。单个图像或整个图像目录可用作代码的输入 。 安装 确保安装了Python3.6环境 克隆库 git clone https://github.com/aqeelanwar/MaskTheFace.git 安装必需的包 cd MaskTheFace pip install –r requirements.txt 运行 cd MaskTheFace # Generic python mask_the_face.py --path