Computer Science

Computer Science ›› 2021, Vol. 48 ›› Issue (9): 168-173.doi: 10.11896/jsjkx.200800001

• Computer Graphics & Multimedia • Previous Articles     Next Articles

Improved YOLOv3 Remote Sensing Target Detection Based on Improved Dense Connection and Distributional Ranking Loss

YUAN Lei1, LIU Zi-yan1, ZHU Ming-cheng1, MA Shan-shan1, CHEN Lin-zhou-ting2   

  1. 1 College of Big Data and Information Engineering,Guizhou University,Guiyang 550025,China
    2 School of Aerospace Engineering,Guizhou Institute of Technology,Guiyang 550003,China
  • Received:2020-08-01 Revised:2020-09-10 Online:2021-09-15 Published:2021-09-10
  • About author:YUAN Lei,born in 1995,postgraduate.His main research interests include deep learning and object detection.
    LIU Zi-yan,born in 1974,associate professor,postgraduate supervisor,is a member of China Computer Federation.Her main research interests include wireless communication system,mobile robot and big data mining analysis.
  • Supported by:
    Science and Technology Foundation of Guizhou Province,China([2016] 1054),Joint Funding Project of Guizhou Province,China([2017]7226),Special Project on the Training and Innovation of New Academic Seedlings of Guizhou University in 2017,China([2017] 5788),Science and Technology Program of Guizhou Province,China([2017] 1069),Major Research Projects on Innovative Groups of Guizhou Provincial Department of Education([2018]026),Engineering Research Center of Ordinary Colleges and Universities of Guizhou Province([2018]007) and Key Projects of Science and Technology Program of Guizhou Province([2019]1416)

PDF (PC)

633

Abstract: Aiming at solving the problems of small object size,uneven sample distribution,and unclear features in remote sensing images,an improved YOLOv3 object detection algorithm is proposed.The Stitcher data enhancement method is used to solve the problem of uneven distribution of small object samples.The VOVDarkNet-53 is proposed.The residual modules of the fourth downsampling in DarkNet-53 are reduced from eight to four.And then the dense connection mode of VOVNet is adopted to extract lower features of small objects to increase the network receptive field.The distributional ranking loss is used to improve the classification loss in YOLOv3 to solve the problem of imbalance between positive and negative samples in single-stage object detector.Comparative experiments are carried out on HRRSD remote sensing datasets by using YOLOv3 object detection algorithm and improved YOLOv3 algorithm.The results demonstrate that the proposed algorithm can achieve better performance of higher detection accuracy of the improved YOLOv3 algorithm for small objects and medium objects are improved by 7.2% and 2.1%,respectively.Although the detection accuracy for large objects is reduced by 1%,the average detection accuracy (mAP) is improved by 4.1%,and the recall and accuracy are also improved.

Key words: Baseline, Object detection, Remote sensing image, Sample imbalance, YOLOv3

CLC Number: 

  • TP391.4
[1]HUANG G Q.Research on Satellite Image Target Detection andrecognition based on Deep Learning[D].Hangzhou:Zhejiang University,2019.
[2]PANG J,LI C,SHI J,et al.R2-CNN:Fast Tiny Object Detection in Large-Scale Remote Sensing Images[J].IEEE Transactions on Geoscience and Remote Sensing,2019,57(8):5512-5524.
[3]SHERMEYER J,VAN ETTEN A.The Effects of Super-Resolution on Object Detection Performance in Satellite Imagery[C]//IEEE Computer Society Conference on Computer Vision and Pattern Recognition Workshops.2019:1432-1441.
[4]RUSSAKOVSKY O,DENG J,SU H,et al.ImageNet LargeScale Visual Recognition Challenge[J].International Journal of Computer Vision,2015,115(3):211-252.
[5]EVERINGHAM M,GOOL L V,WILLIAMS C K I,et al.The Pascal Visual Object Classes (VOC) Challenge[J].International Journal of Computer Vision,2010,88(2):303-338.
[6]LIN T,MAIRE M,BELONGIE S,et al.Microsoft COCO:Common Objects in Context[C]//European Conference on Compu-ter Vision.2014:740-755.
[7]LIU Y,WANG Y,WANG S,et al.CBNet:A Novel Composite Backbone Network Architecture for Object Detection[C]//National Conference on Artificial Intelligence.2020.
[8]TAN M,PANG R,LE Q V,et al.EfficientDet:Scalable and Efficient Object Detection[EB/OL].(2019-11-09) [2020-08-01].https://arxiv.org/abs/1911.09070.pdf.
[9]SONG G,LIU Y,WANG X,et al.Revisiting the Sibling Head in Object Detector[EB/OL].(2020-03-07) [2020-08-01].https://arxiv.org/abs/2003.07540.pdf.
[10]GHIASI G,TAN M,JIN P,et al.SpineNet:Learning Scale-Permuted Backbone for Recognition and Localization[EB/OL].(2019-12-05)[2020-08-01].https://arxiv.org/abs/1912.05027.
[11]LIN T,GOYAL P,GIRSHICK R,et al.Focal Loss for DenseObject Detection[C]//International Conference on Computer Vision.2017:2999-3007.
[12]REDMON J,FARHADI A.YOLOv3:An Incremental Improvement[EB/OL].(2018-04-08) [2020-07-01].https://arxiv.org/abs/1804.02767.
[13]TIAN Z,SHEN C,CHEN H,et al.FCOS:Fully Convolutional One-Stage Object Detection[C]//International Conference on Computer Vision.2019:9627-9636.
[14]LAW H,DENG J.CornerNet:Detecting Objects as Paired Keypoints[C]//European Conference on Computer Vision.2018:765-781.
[15]ZHU W T,XIE B R,WANG M,et al.A review of aircraft target detection in optical remote sensing images[J].Computer Scien-ce,2020,47(S2):165-171.
[16]BELL S,ZITNICK C L,BALA K,et al.Inside-Outside Net:Detecting Objects in Context with Skip Pooling and Recurrent Neural Networks[C]//Computer Vision and Pattern Recognition.2016:2874-2883.
[17]ZHANG M,LI J,DING R L,et al.Research on remote sensing image target detection technology based on improved YOLO-V2 algorithm[J].Computer Science,2020,547 (S1):176-180.
[18]LONG Y,GONG Y,XIAO Z,et al.Accurate Object Localization in Remote Sensing Images Based on Convolutional Neural Networks[J].IEEE Transactions on Geoscience & RemoteSen-sing,2017,55(5):2486-2498.
[19]CHEN Y,ZHANG P,LI Z,et al.Stitcher:Feedback-driven Data Provider for Object Detection[EB/OL].(2020-04-12) [2020-08-01].https://arxiv.org/abs/2004.12432.
[20]LEE Y,PARK J.CenterMask:Real-Time Anchor-Free Instance Segmentation[EB/OL].(2019-11-06) [2020-08-01].https://arxiv.org/abs/1911.06667.pdf.
[21]QIAN Q,CHEN L,LI H,et al.DR Loss:Improving Object Detection by Distributional Ranking[EB/OL].(2019-07-10) [2020-08-01].https://arxiv.org/abs/1907.10156.
[1] WANG Kun-shu, ZHANG Ze-hui, GAO Tie-gang. Reversible Hidden Algorithm for Remote Sensing Images Based on Hachimoji DNA and QR Decomposition [J]. Computer Science, 2022, 49(8): 127-135.
[2] LIU Dong-mei, XU Yang, WU Ze-bin, LIU Qian, SONG Bin, WEI Zhi-hui. Incremental Object Detection Method Based on Border Distance Measurement [J]. Computer Science, 2022, 49(8): 136-142.
[3] WANG Can, LIU Yong-jian, XIE Qing, MA Yan-chun. Anchor Free Object Detection Algorithm Based on Soft Label and Sample Weight Optimization [J]. Computer Science, 2022, 49(8): 157-164.
[4] HAO Qiang, LI Jie, ZHANG Man, WANG Lu. Spatial Non-cooperative Target Components Recognition Algorithm Based on Improved YOLOv3 [J]. Computer Science, 2022, 49(6A): 358-362.
[5] ZHU Wen-tao, LAN Xian-chao, LUO Huan-lin, YUE Bing, WANG Yang. Remote Sensing Aircraft Target Detection Based on Improved Faster R-CNN [J]. Computer Science, 2022, 49(6A): 378-383.
[6] CHEN Yong-ping, ZHU Jian-qing, XIE Yi, WU Han-xiao, ZENG Huan-qiang. Real-time Helmet Detection Algorithm Based on Circumcircle Radius Difference Loss [J]. Computer Science, 2022, 49(6A): 424-428.
[7] CHEN Jia-zhou, ZHAO Yi-bo, XU Yang-hui, MA Ji, JIN Ling-feng, QIN Xu-jia. Small Object Detection in 3D Urban Scenes [J]. Computer Science, 2022, 49(6): 238-244.
[8] HU Fu-yuan, WAN Xin-jun, SHEN Ming-fei, XU Jiang-lang, YAO Rui, TAO Zhong-ben. Survey Progress on Image Instance Segmentation Methods of Deep Convolutional Neural Network [J]. Computer Science, 2022, 49(5): 10-24.
[9] CHU An-qi, DING Zhi-jun. Application of Gray Wolf Optimization Algorithm on Synchronous Processing of Sample Equalization and Feature Selection in Credit Evaluation [J]. Computer Science, 2022, 49(4): 134-139.
[10] XU Tao, CHEN Yi-ren, LYU Zong-lei. Study on Reflective Vest Detection for Apron Workers Based on Improved YOLOv3 Algorithm [J]. Computer Science, 2022, 49(4): 239-246.
[11] GONG Hao-tian, ZHANG Meng. Lightweight Anchor-free Object Detection Algorithm Based on Keypoint Detection [J]. Computer Science, 2021, 48(8): 106-110.
[12] WANG Shi-yun, YANG Fan. Remote Sensing Image Semantic Segmentation Method Based on U-Net Feature Fusion Optimization Strategy [J]. Computer Science, 2021, 48(8): 162-168.
[13] LI Lin, LIU Xue-liang, ZHAO Ye, JI Ping. Low Light Image Fusion Detection Method Based on Lego Filter and SSD [J]. Computer Science, 2021, 48(7): 213-218.
[14] XIN Yuan-xue, SHI Peng-fei, XUE Rui-yang. Moving Object Detection Based on Region Extraction and Improved LBP Features [J]. Computer Science, 2021, 48(7): 233-237.
[15] ZHANG Man, LI Jie, ZHU Xin-zhong, SHEN Ji, CHENG Hao-tian. Augmentation Technology of Remote Sensing Dataset Based on Improved DCGAN Algorithm [J]. Computer Science, 2021, 48(6A): 80-84.
Viewed
Full text


Abstract

Cited
  Shared   
  Discussed   
No Suggested Reading articles found!
渝ICP备16013121号-4
Add:18# Honghu West Rd., Yubei ,Chongqing,China    Post Code: 401121
Tel: 023-63500828/023-67039612/023-67039623
E-mail: jsjkx12@163.com
Powered by Beijing Magtech Co., Ltd.