2024年度  第1期


标题:基于剪枝与知识蒸馏的轻量级人像抠图方法
作者:程航 徐树公
作者单位:上海大学通信与信息工程学院,上海 200444
关键字:神经网络;图像抠图;网络剪枝;知识蒸馏
摘要:近年来,随着人像抠图技术的广泛应用,对其实时性与精度的要求也逐渐提高。现有轻量级方法在精度上难以得到保证,而精度更高的方法往往使用较大的网络结构,无法满足实时性需求。为了解决这一问题,提出了一种基于剪枝与知识蒸馏的轻量级人像抠图方法。该方法首先通过网络剪枝来获得一个轻量级学生网络结构,随后使用该学生网络进行知识蒸馏。实验证明,该方法可以在保证模型精度的前提下有效减少参数量和推理耗时,且相较现有的轻量级人像抠图方法具有更少的参数量和更高的精度。