首页 期刊 激光与光电子学进展 一种多任务的卷积神经网络目标分类算法 【正文】

一种多任务的卷积神经网络目标分类算法

作者:张苗辉; 张博; 高诚诚 河南大学大数据分析与处理河南省重点实验室; 河南开封475004; 河南大学数据与知识工程研究所; 河南开封475004
机器视觉   多任务卷积神经网络   深度学习   细粒度图像  

摘要:提出一种基于细粒度图像和多属性融合的多任务卷积神经网络(MTCNN)。该网络主要包含几个关键环节,首先在网络中增加标签输入层,复制并分离输入的多个标签,通过全连接层与多个任务相匹配,增加与标签数量相应的Softmax损失函数,来对多个任务进行反向传播;然后,使用显著性检测与角点检测相结合的方法,提取出原始图像中的细粒度图像用于MTCNN的数据输入,使神经网络提取到的目标特征具有独特性和区分性;最后,使用非线性激活函数PReLu,进一步提高网络的分类精度。通过在Car Dataset中进行多任务并行训练,测试精度较传统单个任务的分类精度提升10%,实验结果表明,MTCNN有较高的泛化能力,对于图像分类的精度有明显的提升。

注:因版权方要求,不能公开全文,如需全文,请咨询杂志社

学术咨询 免费咨询 杂志订阅