摘要
本发明公开了一种基于边界稳定生成对抗网络模型的高精度头发检测方法,包括:获取头发训练集;建立边界稳定生成对抗网络模型并采用头发训练集和真实头发图像集进行训练;将头发训练集和真实头发图像集再次输入训练好的边界稳定生成对抗网络模型,获得最终风格化图像集;建立旋转目标检测模型;基于均值教师框架采用真实头发图像集、最终风格化图像集和对应的标签集对旋转目标检测模型进行训练,获得训练好的旋转目标检测模型即为均值教师框架的教师模型;利用训练好的均值教师框架的教师模型对待测头发图像进行目标检测,生成每根头发的旋转目标检测框即为头发检测结果。能够提高头发检测任务的灵活性和准确性,适用范围广。
技术关键词
生成对抗网络模型
头发检测方法
图像
教师
风格
框架
采样点
像素点
模块
训练集
画布
学生
标签
颜色
语义
交叉点
矩阵
坐标
无监督
数据