UAV-YOLO: Small Object Detection on Unmanned Aerial Vehicle Perspective.

Mingjie Liu, Xianhao Wang, Anjian Zhou, Xiuyuan Fu, Yiwei Ma, Changhao Piao
Author Information
  1. Mingjie Liu: Department of Automation, Chongqing University of Posts and Telecommunications, No. 2 Chongwen Road, Chongqing 40000, China.
  2. Xianhao Wang: Department of Automation, Chongqing University of Posts and Telecommunications, No. 2 Chongwen Road, Chongqing 40000, China.
  3. Anjian Zhou: Chongqing Changan New Energy Science and Technology Co., Ltd., Chongqing 401120, China.
  4. Xiuyuan Fu: Chongqing SPIC ZINENG Technology Co., Ltd., Chongqing 404100, China.
  5. Yiwei Ma: Department of Automation, Chongqing University of Posts and Telecommunications, No. 2 Chongwen Road, Chongqing 40000, China.
  6. Changhao Piao: Department of Automation, Chongqing University of Posts and Telecommunications, No. 2 Chongwen Road, Chongqing 40000, China.

Abstract

Object detection, as a fundamental task in computer vision, has been developed enormously, but is still challenging work, especially for Unmanned Aerial Vehicle (UAV) perspective due to small scale of the target. In this study, the authors develop a special detection method for small objects in UAV perspective. Based on YOLOv3, the Resblock in darknet is first optimized by concatenating two ResNet units that have the same width and height. Then, the entire darknet structure is improved by increasing convolution operation at an early layer to enrich spatial information. Both these two optimizations can enlarge the receptive filed. Furthermore, UAV-viewed dataset is collected to UAV perspective or small object detection. An optimized training method is also proposed based on collected UAV-viewed dataset. The experimental results on public dataset and our collected UAV-viewed dataset show distinct performance improvement on small object detection with keeping the same level performance on normal dataset, which means our proposed method adapts to different kinds of conditions.

Keywords

References

  1. IEEE Trans Pattern Anal Mach Intell. 2010 Sep;32(9):1627-45 [PMID: 20634557]
  2. Sensors (Basel). 2019 Oct 16;19(20): [PMID: 31623134]
  3. Sensors (Basel). 2020 Mar 27;20(7): [PMID: 32230867]

Word Cloud

Created with Highcharts 10.0.0detectiondatasetsmallUAVperspectivemethodUAV-viewedcollectedobjectObjectUnmannedAerialVehicledarknetoptimizedtwoproposedperformancefundamentaltaskcomputervisiondevelopedenormouslystillchallengingworkespeciallyduescaletargetstudyauthorsdevelopspecialobjectsBasedYOLOv3ResblockfirstconcatenatingResNetunitswidthheightentirestructureimprovedincreasingconvolutionoperationearlylayerenrichspatialinformationoptimizationscanenlargereceptivefiledFurthermoretrainingalsobasedexperimentalresultspublicshowdistinctimprovementkeepinglevelnormalmeansadaptsdifferentkindsconditionsUAV-YOLO:SmallDetectionPerspectiveconvolutionalneuralnetworkunmannedaerialvehicle

Similar Articles

Cited By