face parsing

人脸区域分割

项目介绍

注意：该项目不包括人脸检测部分，人脸检测项目地址：https://codechina.csdn.net/EricLee/yolo_v3

• 图片示例：
  

• 视频示例：
  

项目配置

• 作者开发环境：
• Python 3.7
• PyTorch >= 1.5.1

数据集

• CelebAMask-HQ dataset，数据下载地址：
  https://github.com/switchablenorms/CelebAMask-HQ

• The CelebAMask-HQ dataset is available for non-commercial research purposes only.
• You agree not to reproduce, duplicate, copy, sell, trade, resell or exploit for any commercial purposes, any portion of the images and any portion of derived data.
• You agree not to further copy, publish or distribute any portion of the CelebAMask-HQ dataset. Except, for internal use at a single site within the same organization it is allowed to make copies of the dataset.

• 数据集制作
  下载数据集并解压，然后运行脚本 prepropess_data.py,生成训练用的mask，注意脚本内相关参数配置。

预训练模型

• 预训练模型下载地址(百度网盘 Password: )

项目使用方法

模型训练

• 根目录下运行命令： python train.py (注意脚本内相关参数配置 )

模型推理

• 根目录下运行命令： python inference.py (注意脚本内相关参数配置 )