-
公开(公告)号:CN118229723A
公开(公告)日:2024-06-21
申请号:CN202410250369.1
申请日:2024-03-05
Applicant: 华中科技大学
Abstract: 本申请实施例提供一种基于多任务学习的抠图方法及装置,涉及计算机视觉技术领域,其中方法包括:获取待抠图的原始图像;将原始图像输入至预先训练好的多任务学习模型,获取多任务学习模型输出的透明度遮罩图作为抠图结果;其中,多任务学习模型包括共享编码器、语义分支模块、细节分支模块和基于亲和力的特征传播模块,语义分支模块包括语义分支特征提取模块和语义分支解码器,细节分支模块包括细节分支特征提取模块和细节分支解码器,特征传播模块嵌入语义分支解码器和细节分支解码器之间。本申请实施例在双解码器之间嵌入基于亲和力的特征传播模块,从而缓解传统多任务学习模型潜在的负迁移问题,使得整体模型达到更好的抠图性能。