梯度直方图卷积特征的胶囊网络在交通监控下的车型分类

• 作者: 张睿 张雷 曹建芳 陈立潮
• 来源期刊: 计算机应用
• 学科: 工学
• 关键词: 交通监控 胶囊网络 方向梯度直方图 车型分类 卷积神经网络
• 年，卷（期）: 2020,（10）
• 所属期刊栏目: 人工智能
• 页码范围: 2881-2889
• 页数: 9页
• 分类号: TP391.3
• 语种: 中文
• DOI: 10.11772/j.issn.1001-9081.2020020152

摘要

为了充分利用图像信息以提高现有交通监控下车型分类的效果,在胶囊网络的基础上增加梯度直方图卷积(HOG-C)特征提取方法,提出HOG-C特征的胶囊网络模型——HOG-C CapsNet.首先,使用梯度统计特征提取层对图像中的梯度信息进行统计,构建方向梯度直方图(HOG)特征图;其次,使用卷积层提取出图像的颜色信息,把提取出的颜色信息与HOG特征图融合构成HOG-C特征图;最后,输入卷积层提取HOG-C特征图的抽象特征,并通过胶囊网络对提取的抽象特征进行具有三维空间特征表达的胶囊封装,使用动态路由算法实现车型分类.在BIT-Vehicle数据集上对该模型和其他相关模型进行的对比实验中,该模型得到98.17％的准确率、97.98％的平均精确率均值(MAP)、98.42％的平均召回率均值(MAR)和98.20％的综合评价指标.实验结果表明,该模型在交通监控下的车型分类上具有更好的效果.

作者信息

序号	姓名	单位	发文数	被引次数	H指数	G指数
1	陈立潮		127	682	13.0	19.0
2	张雷		10	19	3.0	4.0
3	张睿		23	41	4.0	5.0
4	曹建芳		12	2	1.0	1.0