Opto-Electronic Engineering, Volume. 51, Issue 5, 240051(2024)
Improved YOLOv7 algorithm for target detection in complex environments from UAV perspective
Figures & Tables(11)
Fig. 4. Comparison of (a) NMS and (b) soft NMS detection effect sample chart
Fig. 6. Visual comparison of YOLOv7 and SSG-YOLOv7 detection effect
Table 1. Anchor frame size generated by K-means++
View tableView in ArticleTable 1. Anchor frame size generated by K-means++
特征图尺寸 感受野 锚框 20x20 Big [33,49],[63,73] 40x40 Medium [14,35],[27,23] 80x80 Small [20,8,8,15,14] 160x160 Tiny [2,5,4,11]
Table 2. Comparison of SPPCSPC and SG-SPPCSPC
View tableView in ArticleTable 2. Comparison of SPPCSPC and SG-SPPCSPC
模块类型 Parameters/M GFLOPS SPPCSPC模块 12.8 16.2 SG-SPPCSPC模块 3.69 4.9
Table 3. Results of ablation experiments
View tableView in ArticleTable 3. Results of ablation experiments
Model K-means++ SimAM SG-SPPCSPC Soft NMS Vis_mAP@0.5/% RSOD_mAP@0.5/% Parameters/M FPS GFLOPs A 40.89 95.60 37.6 82 106.5 B √ 44.15 96.91 37.6 82 106.5 C √ √ 46.40 97.22 37.6 87 107.2 D √ √ √ 48.61 97.91 28.5 93 95.9 E √ √ √ √ 51.34(+10.45) 98.27(+2.67) 28.5 93 95.9
Table 4. Comparison of mAP(%) before and after data enhancement
View tableView in ArticleTable 4. Comparison of mAP(%) before and after data enhancement
Model 原始VisDrone 增强后VisDrone 原始RSOD 增强后RSOD YOLOv7 36.76 40.89 92.01 95.60 SSG-YOLOv7 42.63 51.34 93.82 98.27
Table 5. Comparison of experimental results
View tableView in ArticleTable 5. Comparison of experimental results
Method Visdrone_mAP@0.5 /% Visdrone_mAP@0.5:0.95 /% RSOD_mAP@0.5 /% RSOD_mAP@0.5:0.95 /% FPS Parameters/M Faster R-CNN[ 6 ]20.0 8.91 85.6 54.1 43 137.10 SSD[ 23 ]10.2 5.1 87.4 52.6 249 26.29 YOLOv5s 27.4 15.6 94.0 59.5 126 7.28 YOLOv5m 32.0 18.8 95.2 66.4 98 21.38 YOLOv5l 36.5 21.5 95.1 68.3 75 47.10 YOLOv7[ 13 ]40.8 24.0 95.6 69.6 82 37.62 YOLOv8s 43.1 25.0 94.1 63.0 160 11.17 YOLOv8m 39.6 22.8 94.1 68.7 122 25.90 YOLOv8l 43.7 25.1 96.0 68.9 98 43.69 本文算法 51.3 29.2 98.3 70.0 93 28.49
Tools
Get Citation
Copy Citation Text
Runmei Zhang, Yufei Xiao, Zhennan Jia, Zhong Chen, Zihua Chen, Bin Yuan, Weiwei Cao, Weiwei Song. Improved YOLOv7 algorithm for target detection in complex environments from UAV perspective[J]. Opto-Electronic Engineering, 2024, 51(5): 240051
Paper Information
Category: Article
Received: Mar. 6, 2024
Accepted: Apr. 24, 2024
Published Online: Jul. 31, 2024
The Author Email: Weiwei Song (宋娓娓)
© Copyright 2018-2021 | Chinese Laser Press.
All Rights Reserved 沪ICP备15018463号-20