2023
DOI: 10.3390/rs15133214
|View full text |Cite
|
Sign up to set email alerts
|

Improving YOLOv7-Tiny for Infrared and Visible Light Image Object Detection on Drones

Abstract: To address the phenomenon of many small and hard-to-detect objects in drone images, this study proposes an improved algorithm based on the YOLOv7-tiny model. The proposed algorithm assigns anchor boxes according to the aspect ratio of ground truth boxes to provide prior information on object shape for the network and uses a hard sample mining loss function (HSM Loss) to guide the network to enhance learning from hard samples. This study finds that the aspect ratio difference of vehicle objects under drone pers… Show more

Help me understand this report

Search citation statements

Order By: Relevance

Paper Sections

Select...
1
1
1
1

Citation Types

0
4
0

Year Published

2023
2023
2025
2025

Publication Types

Select...
7

Relationship

0
7

Authors

Journals

citations
Cited by 10 publications
(4 citation statements)
references
References 38 publications
(63 reference statements)
0
4
0
Order By: Relevance
“…Compared to methods dependent on centerline extraction models [18], our proposed reference point positioning technique identifies the location of inter-row navigation lines more accurately. Experimental results indicate that, among other YOLOv7 network model variants [28], our model exhibits the best balance between recognition accuracy and real-time processing capabilities. However, potential false positives and false negatives in the target detection algorithm may still lead to inaccuracies in reference point positioning.…”
Section: Discussionmentioning
confidence: 97%
See 1 more Smart Citation
“…Compared to methods dependent on centerline extraction models [18], our proposed reference point positioning technique identifies the location of inter-row navigation lines more accurately. Experimental results indicate that, among other YOLOv7 network model variants [28], our model exhibits the best balance between recognition accuracy and real-time processing capabilities. However, potential false positives and false negatives in the target detection algorithm may still lead to inaccuracies in reference point positioning.…”
Section: Discussionmentioning
confidence: 97%
“…However, the complexity of the network architecture and the large number of parameters in YOLOv7 network model make it demanding on device performance, rendering it unsuitable for edge terminal devices [26,27]. To address this issue, the researchers designed the YOLOv7-tiny network model based on YOLOv7 network model [28][29][30], which features a simplified structure specifically tailored for edge GPU devices [31]. YOLOv7-tiny network model consists of three components: the backbone network, the neck network, and the prediction head, as illustrated in Fig.…”
Section: Yolov7-tiny Network Modelmentioning
confidence: 99%
“…To address the problem of spatial information loss of small objects, the Spatial Information Enhancement Module (SIEM) is designed to adaptively learn the weak spatial information to be protected for small objects. In [23], anchor boxes are assigned according to the aspect ratio of the ground truth box to provide the network with prior information about the shape of the object. The network uses the Hard Sample Mining Loss (HSM Loss) function to guide learning and provide prior information about the shape of the object.…”
Section: Small Object Detectionmentioning
confidence: 99%
“…While newer versions of YOLO may offer additional advancements, our research concentrated on YOLOv4 to establish a solid foundation for automatic bird detection. Future work could certainly explore the benefits of newer versions, such as YOLOv7 [34] and YOLOv7-tiny [35], considering their specific improvements, but our current focus was to demonstrate the effectiveness of YOLOv4 and YOLOv4-tiny in the context of bird-detection tasks.…”
Section: Future Workmentioning
confidence: 99%