欢迎访问《图学学报》 分享到:

图学学报

• 计算机视觉 • 上一篇    下一篇

基于深度卷积神经网络的多任务细粒度车型识别

  

  1. 1. 福建江夏学院工程学院,福建福州 350108;
    2. 安徽工程大学管理工程学院,安徽芜湖 241000
  • 出版日期:2018-06-30 发布日期:2018-07-10
  • 基金资助:
    福建省中青年教师教育科研项目(JAS160616)

Multitask Fine-Grained Vehicle Identification Based on Deep Convolutional Neural Networks

  1. 1. School of Engineering, Fujian Jiangxia University, Fuzhou Fujian 350108, China;
    2. School of Management Engineering, Anhui Polytechnic University, Wuhu Anhui 241000, China
  • Online:2018-06-30 Published:2018-07-10

摘要: 车型识别,尤其是细粒度车型识别是现代智能交通系统的重要组成部分。针对传
统车型识别方法难以进行有效的细粒度车型识别的问题,以AlexNet、GoogleNet 及ResNet 等3
种经典深度卷积神经网络架构作为基础网络,引入了车辆的类型分类作为辅助任务,从而与细
粒度车型识别任务一起构成了一个多任务联合学习的模型。通过在一个包含281 个车型类别的
公开数据集上对模型进行训练及测试,在无需任何车辆的部件位置标注及额外的3D 信息的情
况下,验证了该模型在在细粒度车型识别任务上表现出的优异性能,同时多任务学习策略的引
入可使得模型性能相比任一单任务学习时的性能均有所提高,最终实现了一个简洁高效的细粒
度车型识别模型,基本满足实际应用需求。

关键词: 智能交通, 细粒度车型识别, 深度学习, 神经网络

Abstract: Vehicle identification, especially fine-grained vehicle identification, is an important part of
modern intelligent transportation system. Aiming at the problem that it is difficult to effectively
recognize fine-grained vehicle using traditional vehicle identification methods, we take three classic
deep convolutional neural networks (such as AlexNet, GoogleNet and ResNet) as the basic networks,
and introduce the classification of vehicle types as the auxiliary task, together with fine-grained
vehicle identification task to constitute a multitask joint-learning model. By training and evaluating
our model on a public data set which contains 281 vehicle types, we have demonstrated the excellent
performance of this model in fine-grained vehicle identification task with no need of annotations
about vehicle parts’ location and additional 3D information. Besides, with the introduction of
multitask learning strategy, the performance of this model can be improved, compared with that of
any single-task learning model. Our model is simple and efficient, and can basically meet the demand
of practical applications.

Key words: intelligent transportation, fine-grained vehicle identification, deep learning, neural
networks