[发明专利]一种基于卷积神经网络的图像头发识别方法及其装置

摘要

本发明实施例公开了一种基于卷积神经网络的图像头发识别方法及其装置，其中，该方法包括：收集图集；对图集中的每一张图片进行标记；对图集中的每一张图片进行预处理；检测图集中每一张图片的人像头部位置，获得训练图集对应的头部位置信息和测试图集对应的头部位置信息；基于全卷积网络构造头发全卷积网络；对头发全卷积网络进行训练；将输出的结果与头发区域遮蔽测试图集相比较，获得对头发全卷积网络的评估结果；将欲获取头发遮蔽图的图片输入头发全卷积网络，获得该图像头发区域的遮蔽图。可以解决现有技术中难以处理背景颜色与头发颜色相似的情况，以及无法在图像人脸偏转角度很大或人背面的情况下无法识别头发的问题。

主权项

1.一种基于卷积神经网络的图像头发识别方法，其特征在于，所述方法包括：/n收集图集，该图集包括训练图集和测试图集；/n对图集中的每一张图片进行标记，获得训练图集对应的头发区域遮蔽训练图集和测试图集对应的头发区域遮蔽测试图集；/n对图集中的每一张图片进行预处理，获得与训练图集对应的YCrCb训练图集和频率遮蔽训练图集、与测试图集对应的YCrCb测试图集和频率遮蔽测试图集；/n检测图集中每一张图片的人像头部位置，获得训练图集对应的头部位置信息和测试图集对应的头部位置信息；/n基于全卷积网络构造头发全卷积网络HFCN；/n将YCrCb训练图集、频率遮蔽训练图集及训练图集的头部位置信息进行编码，输入到头发全卷积网络，同时，将头发区域遮蔽训练图集输入头发全卷积网络，对头发全卷积网络进行训练，获得训练好的头发全卷积网络；/n将YCrCb测试图集、频率遮蔽测试图集及测试图集的头部位置信息进行编码，并输入到训练好的头发全卷积网络，将输出的结果与头发区域遮蔽测试图集相比较，获得对头发全卷积网络的评估结果；/n将欲获取头发遮蔽图的图片输入头发全卷积网络，获得该图像头发区域的遮蔽图。/n