SS-YOLOv8: small-size object detection algorithm based on improved YOLOv8 for UAV imagery

计算机科学 计算机视觉 人工智能 密码学 目标检测 对象(语法) 计算机图形学 计算机图形学(图像) 算法 模式识别(心理学)
作者
Jinlong Qu,Qi Li,Jie Pan,Mingzheng Sun,Xingzheng Lu,Ying Zhou,Hongliang Zhu
出处
期刊:Multimedia Systems [Springer Science+Business Media]
卷期号:31 (1) 被引量:19
标识
DOI:10.1007/s00530-024-01622-3
摘要

Unmanned aerial vehicle (UAV) image object detection has extensive applications across both civilian and military domains. However, the traditional YOLOv8 detection algorithm faces significant challenges in detecting small objects in UAV imagery, primarily due to a high missed detection rate and an excessive number of parameters. To address these issues, this paper introduces an enhanced small object detection approach, called Small-Size Object Detection Algorithm Based on Improved YOLOv8 for UAV Imagery (SS-YOLOv8). Firstly, considering the difficulties and stringent real-time requirements in detecting small objects in UAV aerial images, this work streamlines the model by eliminating the large object detection head and its associated redundant network layers, significantly limiting the parameter count. Furthermore, a specialized tiny object detection head is proposed. It is custom-designed for detecting small objects. To enhance the model’s capacity for extracting fine features of small objects and mitigate information loss during feature extraction, the convolution module is replaced in the backbone with space-to-depth convolution (SPD-Conv). Moreover, in the neck section, the self created GCU module is incorporated as it effectively amalgamates deep semantic features with shallow positional ones. Finally, while maintaining consistent parameter costs, multi-scale features are meticulously reused and incorporated to achieve a more comprehensive and sophisticated feature fusion. The experimental results on the VisDrone2019 dataset showed that compared with YOLOv8, SS-YOLOv8 improved its MAP index by 6.9% on the validation set and 5.8% on the test set, while reducing the number of parameters and model size by 65.93% and 64.85%, respectively. Compared with the best performing comparison algorithm, SS-YOLOv8 has the highest detection accuracy, indicating the superiority of this method.
最长约 10秒,即可获得该文献文件

科研通智能强力驱动
Strongly Powered by AbleSci AI
科研通是完全免费的文献互助平台,具备全网最快的应助速度,最高的求助完成率。 对每一个文献求助,科研通都将尽心尽力,给求助人一个满意的交代。
实时播报
lucygaga发布了新的文献求助10
刚刚
脑洞疼应助背后的若之采纳,获得10
刚刚
1秒前
2秒前
3秒前
3秒前
3秒前
加减乘除发布了新的文献求助30
3秒前
3秒前
3秒前
缥缈傥完成签到,获得积分10
3秒前
JoJoT完成签到,获得积分20
4秒前
ni完成签到 ,获得积分10
5秒前
6秒前
6秒前
所所应助幸福雪青采纳,获得10
6秒前
6秒前
7秒前
畸你太美发布了新的文献求助100
7秒前
shisan发布了新的文献求助10
7秒前
络梦摘星辰完成签到,获得积分10
8秒前
愿518发布了新的文献求助10
8秒前
weie完成签到,获得积分10
8秒前
MIST发布了新的文献求助10
8秒前
shuimenw完成签到,获得积分10
9秒前
9秒前
空心胶囊完成签到,获得积分10
10秒前
天天向上完成签到,获得积分20
10秒前
10秒前
zz发布了新的文献求助10
10秒前
10秒前
health__up完成签到,获得积分10
11秒前
希望天下0贩的0应助JW.Huang采纳,获得10
12秒前
小鹏哥完成签到,获得积分10
12秒前
领导范儿应助doudou采纳,获得10
12秒前
13秒前
LDL完成签到,获得积分10
13秒前
周围完成签到,获得积分10
14秒前
乾之三爻发布了新的文献求助10
14秒前
牛马中的老白菜完成签到,获得积分10
15秒前
高分求助中
Principles of Economics, 11th Edition 10000
University Physics with Modern Physics, 16th edition 10000
(应助此贴封号)【重要!!请各用户(尤其是新用户)详细阅读】【科研通的精品贴汇总】 10000
Molecular Mechanisms of Photosynthesis, 4th Edition 1000
Organic Reactions, Volume 116 1000
Matrix Methods in Data Mining and Pattern Recognition 510
Social Skills Improvement System-Rating Scales--Chinese Version 500
热门求助领域 (近24小时)
化学 材料科学 医学 生物 纳米技术 工程类 有机化学 化学工程 生物化学 计算机科学 内科学 物理 复合材料 催化作用 细胞生物学 无机化学 光电子学 物理化学 电极 基因
热门帖子
关注 科研通微信公众号,转发送积分 7254586
求助须知:如何正确求助?哪些是违规求助? 8876687
关于积分的说明 18742738
捐赠科研通 6935086
什么是DOI,文献DOI怎么找? 3200159
关于科研通互助平台的介绍 2374831
邀请新用户注册赠送积分活动 2175117