摘要
目前的肖像绘画机器人普遍存在画像质量较低、艺术风格单一、绘制时间较长、运算复杂度高等问题,针对这些问题,提出了一种基于知识蒸馏的简约人脸线条画生成方法。首先,使用大模型作为教师网络,通过知识蒸馏思想,提出一种蒸馏损失来指导轻量型学生网络的训练过程,从而使轻量型模型达到较好的生成效果。其次,引入风格损失,提升了生成线条的连续性。实验结果表明:该方法可以生成简洁、美观的人脸线条画,并且机器人能够在2 min内绘制完成。相比于教师模型计算量减少了约65.3%,参数量减少了约97.1%,且显著优于基准学生模型。
-
单位杭州电子科技大学; 高等研究院; 国网山东省电力公司烟台供电公司