一种抠图模型训练、图像抠图的方法及装置
摘要:
本申请适用于图像处理的技术领域,提供一种抠图模型训练、图像抠图的方法及装置,所述校正方法包括:获取训练样本集合、初始教师模型以及初始学生模型;通过所述训练样本集合,训练所述初始教师模型得到目标教师模型以及所述目标教师模型输出的第一透明度蒙版;分别将所述目标教师模型中的第一权重参数迁移至所述初始学生模型中的各个子网络中,得到过渡学生模型;根据所述第一透明度蒙版以及所述训练样本集合,训练所述过渡学生模型得到抠图模型。由于抠图模型不仅拥有目标教师模型中的第一权重参数,且不断学习目标教师模型输出的第一透明度蒙版,故在保证处理精度的前提下,巧妙地缩减了模型的体量以及处理时长。
公开/授权文献
0/0