基于多任务深度神经网络的图像情感区域检测

作者:徐晶晶; 张俊升; 李腾腾; 祝汉城; 谢涛; 余伟*
来源:计算机工程与设计, 2020, 41(09): 2616-2622.
DOI:10.16208/j.issn1000-7024.2020.09.032

摘要

为得到准确的图像情感区域,设计一种基于多任务深度神经网络的图像情感区域检测方法。提出的网络包括两个任务:图像情感区域检测任务和图像情感分类任务,两个任务的共享层为VGG-16网络结构,图像的情感区域检测任务和情感分类任务通过对共享层参数的训练可以提高单个任务的泛化能力,提升网络的性能。实验结果表明,该方法可以准确检测图像的情感区域,检测性能优于单任务情感区域检测方法和现有的显著性检测和物体检测方法。

全文