基于YOLOv8的无人机图像目标检测算法

文章预览

作者：赵继达，甄国涌，储成群 来源：《计算机工程》 编辑：东岸因为@一点人工一点智能 原文：http://www.ecice06.com/CN/10.19678/j.issn.1000-3428.0068268#1 在无人机(UAV)目标检测任务中，存在因检测目标尺度小、检测图像背景复杂等原因导致的漏检、误检问题。针对上述问题，提出改进YOLOv8s的无人机图像目标检测算法。 首先，针对无人机拍摄目标普遍为小目标的应用场景，减少算法骨干网络(Backbone)层数，增大待检测特征图尺寸，使得网络模型更专注于微小目标；其次，针对数据集普遍存在一定数量低质量示例影响训练效果的问题，引入Wise-IoU损失函数，增强数据集训练效果；再次，通过引入上下文增强模块，获得小目标在不同感受野下的特征信息，改善算法在复杂环境下对小目标的定位和分类效果；最后，设计空间-通道滤波模块，增强卷积过程中目标的 ………………………………