计算机应用

作者：陈立潮, 张雷, 曹建芳, 张睿

单位：1. 太原科技大学 计算机科学与技术学院, 太原 030024;2. 忻州师范学院 计算机系, 山西 忻州 034000

关键词：交通监控,胶囊网络,方向梯度直方图,车型分类,卷积神经网络

基金：山西省应用基础研究项目（201801D221179）。

为了充分利用图像信息以提高现有交通监控下车型分类的效果，在胶囊网络的基础上增加梯度直方图卷积（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%的综合评价指标。实验结果表明，该模型在交通监控下的车型分类上具有更好的效果。

来源：2020年第10期

《计算机应用》期刊编辑部

查看计算机应用杂志2020年第10期