轻量型简约人脸线条画生成方法

作者:阮杰; 蒋畅; 朱静洁; 戴玲娜; 李荣生; 高飞*; 李鹏
来源:西安工程大学学报, 2022, 36(05): 45-52.
DOI:10.13338/j.issn.1674-649x.2022.05.007

摘要

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

全文