首页|基于双流对比特性学习和图像多尺度退化增强的小目标检测方法

基于双流对比特性学习和图像多尺度退化增强的小目标检测方法

扫码查看
针对小目标检测任务中目标图像尺寸小、目标特征信息模糊、目标和背景难区分等问题,提出一种基于双流对比特性学习和图像多尺度退化增强的小目标检测方法.首先,将对比学习模型的输入图像进行多尺度退化增强,增强算法对小目标的捕获感知;其次,在空间域和频率域同时进行对比学习表征,以学习更具鉴别性的目标识别特征,增强模型对目标与背景的区分能力,从而提高小目标检测的效果.为验证所提方法的有效性设计了消融实验,并对比分析了与其他先进算法的检测性能优劣.实验结果表明:所提方法在MS COCO数据集上平均精度均值mAP相较基线算法提升3.6个百分点,小目标平均精度均值mAPS相较主流先进算法提升7.7个百分点;在VisDrone2019数据集上,所提方法平均精度均值mAP较基线算法提升2.4个百分点,所提方法综合性能优于基线算法与其他主流先进算法.可视化检测效果分析表明,所提方法在小目标检测上的漏检、误检问题得到较大改善.
Small object detection based on dual-stream contrastive feature learning and multi-scale image degradation augmentation
To address the challenges in small object detection tasks,such as the small size of target images,blurred target features,and difficulty in distinguishing targets from backgrounds,a method based on dual-stream contrastive feature learning and multi-scale image degradation augmentation is proposed.First,the input images of the contrastive learning model are subjected to multi-scale degradation augmentation,thus enhancing the model's ability to perceive and capture small targets.Second,contrastive learning representations are conducted in both spatial and frequency domains simultaneously to learn more discriminative target recognition features,thereby improving the model's ability to differentiate between targets and backgrounds.To verify the effectiveness of the proposed scheme,ablation experiments are designed,and the detection performance is compared with that of other advanced algorithms.Experimental results show that the proposed scheme achieves an improvement of 3.6%in mean Average Precision(mAP)over the baseline algorithm on the MS COCO dataset,and an improvement of 7.7%in mAP for small objects(mAPs)compared to mainstream advanced algorithms.On the VisDrone2019 dataset,the proposed method achieves a 2.4%increase in mAP compared to the baseline algorithm,demonstrating its superior overall performance over the baseline algorithm and other mainstream advanced algorithms.Visual analysis of detection re-sults indicates a significant improvement in the rates of false negatives and false positives for small object detection.

small object detectioncontrastive learningdual-stream networkimage degradationimage enhance-mentmultiscale

王宇、何志、康朋新、涂晓光、周超、刘建华、雷霞、王文敬

展开 >

中国民用航空飞行学院航空电子电气学院,广汉,618307

西南技术物理研究所,成都,610041

四川大学计算机学院,成都,610065

中国民用航空飞行学院四川省通用航空器维修工程技术研究中心,广汉,618307

展开 >

小目标检测 对比学习 双流网络 图像退化 图像增强 多尺度

2024

南京信息工程大学学报
南京信息工程大学

南京信息工程大学学报

CSTPCD北大核心
影响因子:0.737
ISSN:1674-7070
年,卷(期):2024.16(6)