Object Detection Algorithm Based on Improved YOLOv7 for UAV Images

Yi Hsiu Chung*, Chung Yen Su

*Corresponding author for this work

Research output: Chapter in Book/Report/Conference proceedingConference contribution

Abstract

The technology of unmanned aerial vehicles (UAVs) has matured and widened the range of applications in engineering, agriculture, transportation, surveillance, and so on. Deep learning techniques for object detection assist operators in identifying targets and enhancing efficiency. Thus, it is important to improve the accuracy of object detection. However, due to the limited resources of the UAV platform and the necessity for real-time recognition, striking a balance between accuracy enhancement and decreasing computation is pivotal for development. Thus, we proposed an algorithm based on a YOLOv7 single-stage object detector with a detection head to improve the detection effect of small objects and using a modified feature layer attention module (M-FLAM) at the high-level feature layer to enhance the attention on small objects. The algorithm employed modified efficient layer aggregation network (M-ELAN) to reduce the number of parameters without significant loss in accuracy. Experiments were conducted on the VisDrone dataset. YOLOv7 achieved mean average precision (mAP) @ 0.5 of 46.6%, mAP @ 0.5:0.95 of 26.2%, and the number of parameters was 37.2 million. In comparison to YOLOv7, the developed algorithm demonstrated a 2.3% increase in mAP @ 0.5 and a 2.3% increase in mAP @ 0.5:0.95, with a notable 4.8% reduction in the number of parameters.

Original languageEnglish
Title of host publication2023 IEEE 5th Eurasia Conference on IOT, Communication and Engineering, ECICE 2023
EditorsTeen-Hang Meen
PublisherInstitute of Electrical and Electronics Engineers Inc.
Pages18-21
Number of pages4
ISBN (Electronic)9798350314694
DOIs
Publication statusPublished - 2023
Event5th IEEE Eurasia Conference on IOT, Communication and Engineering, ECICE 2023 - Yunlin, Taiwan
Duration: 2023 Oct 272023 Oct 29

Publication series

Name2023 IEEE 5th Eurasia Conference on IOT, Communication and Engineering, ECICE 2023

Conference

Conference5th IEEE Eurasia Conference on IOT, Communication and Engineering, ECICE 2023
Country/TerritoryTaiwan
CityYunlin
Period2023/10/272023/10/29

Keywords

  • UAV
  • Yolov7
  • object detection

ASJC Scopus subject areas

  • Artificial Intelligence
  • Computer Networks and Communications
  • Computer Science Applications
  • Computer Vision and Pattern Recognition
  • Hardware and Architecture
  • Human-Computer Interaction
  • Media Technology

Fingerprint

Dive into the research topics of 'Object Detection Algorithm Based on Improved YOLOv7 for UAV Images'. Together they form a unique fingerprint.

Cite this