首页|基于改进YOLOv8的定向边界框目标检测模型

基于改进YOLOv8的定向边界框目标检测模型

Oriented Bounding Box Object Detection Model Based on Improved YOLOv8

扫码查看
在高分辨率遥感图像的定向边界框(OBB)目标检测研究中,由于目标太小且方向不同,会出现漏检和错检的问题.现有的遥感图像定向边界框目标检测方法虽然取得了不错的进展,但主要侧重于方向建模,而较少考虑目标的大小以及漏检问题.本研究提出了一种基于改进YOLOv8的遥感图像定向物体检测方法.该方法可以提高遥感图像中定向物体的检测精度.首先,创新性地设计了一个ResCBAMG模块,可以更好地提取通道和空间相关信息.其次,提出了一种创新的自上而下的特征融合层网络结构,与ECA注意力模块相结合,有助于适当捕捉局部间的跨通道交互信息.最后,在自下而上特征融合层的不同C2f模块和检测头之间引入了创新的ResCBAMG模块.这种创新结构有助于模型更好地聚焦目标区域,同时也能提高定向目标检测的准确性和鲁棒性.在DOTA-v1.5数据集上的实验结果表明,与原始模型相比,本研究改进模型的检测精度、mAP@0.5和mAP@0.5:0.95指标均有所提高,这也有效改善了对小型目标和复杂场景的检测能力.
In the study of oriented bounding boxes(OBB)object detection in high-resolution remote sensing images,the problem of missed and wrong detection of small targets occurs because the targets are too small and have different orientations.Existing OBB object detection for remote sensing images,although making good progress,mainly focuses on directional modeling,while less consideration is given to the size of the object as well as the problem of missed detection.In this study,a method based on improved YOLOv8 was proposed for detecting oriented objects in remote sensing images,which can improve the detection precision of oriented objects in remote sensing images.Firstly,the ResCBAMG module was innovatively designed,which could better extract channel and spatial correlation information.Secondly,the innovative top-down feature fusion layer network structure was proposed in conjunction with the Efficient Channel Attention(ECA)attention module,which helped to capture inter-local cross-channel interaction information appropriately.Finally,we introduced an innovative ResCBAMG module between the different C2f modules and detection heads of the bottom-up feature fusion layer.This innovative structure helped the model to better focus on the target area.The precision and robustness of oriented target detection were also improved.Experimental results on the DOTA-v1.5 dataset showed that the detection Precision,mAP@0.5,and mAP@0.5:0.95 metrics of the improved model are better compared to the original model.This improvement is effective in detecting small targets and complex scenes.

Remote sensing imageOriented bounding boxes object detectionSmall target detectionYOLOv8

赵新康、司占军

展开 >

天津科技大学 人工智能学院,天津 300457

遥感图像 定向边界框目标检测 小目标检测 YOLOv8

2024

数字印刷
中国印刷科学技术研究所

数字印刷

北大核心
ISSN:2095-9540
年,卷(期):2024.(4)