Improved YOLOv5s for Small Vehicle Object Detection on Remote Sensing Image

Abstract

Abstract: Because of its good detection effect and low computational complexity， YOLOv5s is widely used in various target detection tasks. However， its large downsampling stride makes it difficult to obtain satisfactory results for small-sized vehicle detection in satellite remote sensing images. In order to improve the performance of small target detection， on the basis of YOLOv5s， the strategy of reducing the downsampling stride is adopted to protect the texture and geometric features of small targets in the vehicle， and the attention mechanism module is inserted in front of the detection head to suppress the interference of complex background. The tested results on the autonomous data set with a resolution of 0.5 m/pixel show that the AP， recall and precision of SA-YOLOv5s for vehicle target detection reached 94.1%， 99% and 87.3% respectively， which were 16.4， 6 and 5 percentage points higher than YOLOv5s， and showed good detection performance.

Key words: remote sensing image, small object detection, YOLOv5, autonomous dataset

QIU Di-fa, YU Shu-fang, LIU Jin-hui, BI Meng-zhao. Improved YOLOv5s for Small Vehicle Object Detection on Remote Sensing Image[J]. Computer and Modernization, 2023, 0(05): 122-126.

References

［1］ LIN T Y， MAIRE M， BELONGIE S， et al. Microsoft COCO: Common objects in context［C］// European Conference on Computer Vision. Springer， 2014:740-755.
［2］ EVERINGHAM M， GOOL L， WILLIAMS C K， et al. The pascal visual object classes （VOC） challenge［J］. International Journal of Computer Vision， 2010，88（2）:303-338.
［3］ RUSSAKOVSKY O， DENG J， SU H， et al. ImageNet large scale visual recognition challenge［J］. International Journal of Computer Vision， 2015，115（3）:211-252.
［4］ CHENG M H， LI X Q， ZHU F X， et al. Application of remote sensing big data technology in refined urban management［C］// 2020 International Conference on Big Data， Artificial Intelligence and Internet of Things Engineering （ICBAIE）. 2020:397-400.
［5］ SIMONYAN K， ZISSERMAN A. Very deep convolutional networks for large-scale image recognition［J］. arXiv preprint arXiv:1409.1556， 2014.
［6］ KRIZHEVSKY A， SUTSKEVER I， HINTON G E. ImageNet classification with deep convolutional neural networks［J］. Communications of the ACM， 2017，60（6）:84-90.
［7］ SZEGEDY C， LIU W， JIA Y Q， et al. Going deeper with convolutions［C］// Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition. 2015:1-9.
［8］ GIRSHICK R， DONAHUE J， DARRELL T， et al. Rich feature hierarchies for accurate object detection and semantic segmentation［C］// Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition. 2014:580-587.
［9］ GIRSHICK R. Fast R-CNN［C］// Proceedings of the IEEE International Conference on Computer Vision. 2015:1440-1448.
［10］ REN S Q， HE K M， GIRSHICK R， et al. Faster R-CNN: Towards real-time object detection with region proposal networks［J］. IEEE Transactions on Pattern Analysis & Machine Intelligence， 2017，39（6）:1137-1149.
［11］ REDMON J， DIVVALA S， GIRSHICK R， et al. You only look once: Unified， real-time object detection［C］// Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition. 2016:779-788.
［12］ REDMON J， FARHADI A. YOLO9000: Better， faster， stronger［C］// Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition. 2017:7263-7271.
［13］ REDMON J， FARHADI A. YOLOV3: An incremental improvement［J］. arXiv preprint arXiv:1804.02767， 2018.
［14］ BOCHKOVSKIY A， WANG C， LIAO H M. YOLOv4: Optimal speed and accuracy of object detection［J］. arXiv preprint arXiv:2004.10934， 2020.
［15］ LI C Y， LI L L， JIANG H L， et al. YOLOv6: A single-stage object detection framework for industrial applications［J］. arXiv preprint arXiv:2209.02976， 2022.
［16］ WANG C Y， BOCHKOVSKIY A， LIAO H M. YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors［J］. arXiv preprint arXiv:2207.
02696， 2022.
［17］ LIU W， ANGUELOV D， ERHAN D， et al. SSD: Single shot multibox detector［C］// European Conference on Computer Vision. Springer， 2016:21-37.
［18］ FU C Y， LIU W， RANGA A， et al. DSSD: Deconvolutional single shot detector［J］. arXiv preprint arXiv:1701.06659， 2017.
［19］ LIN T Y， DOLLAR P， GIRSHICK R， et al. Feature pyramid networks for object detection［C］ // Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition. 2017:2117-2125.
［20］ RABBI J， RAY N， SCHUBERT M， et al. Small-object detection in remote sensing images with end-to-end edge-enhanced GAN and object detector network［J］. Remote Sensing， 2020，12（9）:1432-1459.
［21］ GOODFELLOW I J， POUGET-ABADIE J， MIRZA M， et al. Generative adversarial networks［J］. arXiv preprint arXiv:1406.2661， 2014.
［22］ LIM J S， ASTRID M， YOON H J， et al. Small object detection using context and attention［C］// International Conference on Artificial Intelligence in Information and Communication （ICAIIC）. 2021:181-186.
［23］高新波，莫梦竟成，汪海涛，等. 小目标检测研究进展［J］. 数据采集与处理， 2021，36（3）:391-417.
［24］ YUN S， HAN D Y， OH S J， et al. Cutmix: Regularization strategy to train strong classifiers with localizable features［C］// Proceedings of the IEEE/CVF International Conference on Computer Vision. 2019:6023-6032.
［25］ LIU S， QI L， QIN H F， et al. Path aggregation network for in-stance segmentation［C］// Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition. 2018:8759-8768.
［26］ WOO S， PARK J， LEE J Y， et al. CBAM: Convolutional block attention module［C］// Proceedings of the European Conference on Computer Vision （ECCV）. 2018:3-19.
［27］ ZHU X K， LYU S C， WANG X， et al. TPH-YOLOv5s: Improved YOLOv5s based on transformer prediction head for object detection on drone-captured scenarios［C］// IEEE/CVF International Conference on Computer Vision Workshops （ICCVW）. 2021:2778-2788.
［28］ GE Z， LIU S T， WANG F， et al. Yolox: Exceeding YOLO series in 2021［J］. arXiv preprint arXiv:2107.08430， 2021.

[1]	LIU Baobao, YANG Jingjing, TAO Lu, WANG Heying . DSMSC Based on Attention Mechanism for Remote Sensing Image Scene Classification [J]. Computer and Modernization, 2024, 0(12): 72-77.
[2]	ZHENG Shangpo1, CHEN Defu1, LI Jianli2, LIN Guoxian2, WANG Xingping3. Pedestrian Tracking Algorithm Based on Improved YOLOv5s and DeepSORT [J]. Computer and Modernization, 2024, 0(08): 54-58.
[3]	WANG Tao1, 2, HUANG Dan1, 2, LIU Chanyi1, 2, ZHU Tao1, 2. Vehicle Detection in UAV Image Based on YOLOv5s [J]. Computer and Modernization, 2024, 0(08): 108-113.
[4]	PANG Mei, WANG Gong, ZHAN Yong, HUANG Zhefa. Underwater Trash Detection Method Based on Improved YOLOv5 [J]. Computer and Modernization, 2024, 0(07): 120-126.
[5]	WANG Zhibo, MA Han, FENG Jinliang, LIU Guoming. Helmet Detection Algorithm Based on CE-YOLOv5s [J]. Computer and Modernization, 2024, 0(04): 55-59.
[6]	XUE Yang-yi1, ZHOU Li-fan2, GONG Sheng-rong1, 2. Few-shot Algorithm for Object Detection in Remote Sensing Images [J]. Computer and Modernization, 2024, 0(02): 43-49.
[7]	YANG Ke, DONG Bing, WU Yue, HAO Kuan-gong, PENG Zi-chen. Lightweight YOLOv4-based Target Detection Method for Remote Sensing Images of#br# Airport Fields [J]. Computer and Modernization, 2024, 0(02): 93-99.
[8]	ZHANG Jia-Qi, XU Qi-lei. Apple Defect Detection Algorithm Based on NAM-YOLO Network [J]. Computer and Modernization, 2023, 0(10): 53-58.
[9]	XU Ye-tong, GENG Xin-zhe, ZHAO Wei-qiang, ZHANG Yue, NING Hai-long, LEI Tao. A Remote Sensing Image Change Detection Model Based on CNN-Transformer Hybrid Structure [J]. Computer and Modernization, 2023, 0(07): 79-85.
[10]	GONG Xuan, GUO Zhong-hua, CHEN Wang. Remote Sensing Image Road Segmentation Based on CA-TransUNet [J]. Computer and Modernization, 2023, 0(07): 112-118.
[11]	LUO Wei, LIU Si-yuan, XU Jian-xiang, DONG Tian-pei. Improved Solar Cell Defect Detection Algorithm Based on YOLOv5s [J]. Computer and Modernization, 2023, 0(07): 119-126.
[12]	WANG Yi-cheng, ZHANG Guo-liang, ZHANG Zi-jie, . Small Object Detection Method Based on Improved YOLOv5 [J]. Computer and Modernization, 2023, 0(05): 100-105.
[13]	ZHU Li-qing, LI Xiang, . Vehicle Detection of Remote Sensing Images Based on Improved YOLOv5 Algorithm [J]. Computer and Modernization, 2023, 0(05): 117-121.
[14]	TANG Hao-wei, YAO Jun-cai, YAO Cong-ying, SUN Ying, PEI Xing-yi, SONG Chun-xiao. Multi-target Detection of Transmission Lines Based on Improved YOLOv5 [J]. Computer and Modernization, 2023, 0(02): 78-82.
[15]	WANG Hong-yi, KONG Mei-mei, XU Rong-qing. Flame Detection Algorithm Based on Improved YOLOV5 [J]. Computer and Modernization, 2023, 0(01): 103-107.