• •    

基于卷积神经网络的航空器检测与识别

俞汝劼1,熊惠霖2   

  1. 1. 上海交通大学
    2. 上海交通大学电子信息与电气工程学院自动化系
  • 收稿日期:2016-10-12 修回日期:2016-12-08 发布日期:2016-12-08
  • 通讯作者: 俞汝劼

Aircraft Detection and Recognition Framework Based on Deep Convolutional Neural Networks

Ru-Jie YU,   

  • Received:2016-10-12 Revised:2016-12-08 Online:2016-12-08
  • Contact: Ru-Jie YU

摘要: 本文针对军用机场大尺寸卫星图像中航空器检测识别的具体应用场景,建立了一套实时目标检测识别框架,将深度卷积神经网络应用到大尺寸图像中的航空器目标检测与识别任务上。首先,将目标检测的任务看成空间上独立的bounding-box的回归问题,用24层卷积神经网络来完成bouding-box的预测;然后,利用图像分类网络来完成目标切片的分类任务。大尺寸图像上的传统目标检测识别算法通常在时间效率上很难突破,而基于卷积神经网络的航空器目标检测识别算法充分利用了计算硬件的优势,大大缩短了任务耗时。在符合应用场景的自采数据集上进行测试,目标检测实时性达到平均5.765秒/张,在召回率(recall)65.1%的情况下精确度(precision)为79.2%,分类网络的实时性达到平均0.972秒/张,top-1错误率为13%。

关键词: 深度学习, 卷积神经网络, 航空器检测, 目标检测识别, 图像分类

Abstract: An end-to-end framework based on deep convolutional neural networks which focuses on implementing both detection and recognition tasks of some specific object on large-scale pictures was proposed. Firstly, it frames object detection as a regression problem to spatially separated bounding boxes, and predicts bounding boxes based on a 24-layer convolutional neural network. After solving the regression problem of bounding boxes, a sub-framework was presented which could be implemented with several typical classification networks. Such framework was applied to efficiently detect and recognize aircraft in large-scale images which might be token by remote sensing satellites. Since taking full advantage of the powerful GPU, experiments on private dataset which meets the specific application scenario showed that the efficiency of detection net speed up a lot compared to classical algorithms.

Key words: deep learning, convolutional neural networks, aircraft detection, object detection and recognition, image classification

中图分类号: