一种抠图网络训练方法及抠图方法

摘要

本申请涉及计算机视觉技术领域，尤其涉及一种抠图网络训练方法及抠图方法。该方法包括：获取训练样本集合和初始网络；训练样本集合包括多个训练样本，每个训练样本包括输入图像样本，输入图像样本包括具备前景的待抠图图像样本、背景图像样本以及前景的软分割样本，软分割样本利用待抠图图像样本对应的深度图像减去背景图像样本对应的深度图像生成；初始网络包括至少一个阶段网络；阶段网络包括串联的环境组合模块、主干区块和预测模块；利用训练样本集合，训练初始网络得到抠图网络。本申请实施例采用轻量化的模型，实现了高精度的前景分割。