基于改进EfficientDet的雪豹红外相机图像检测方法

摘要：

针对红外相机图像中雪豹存在难检测、难辨认等问题,提出了一个基于域迁移和新型注意力机制的EfficientDet雪豹检测算法.该算法首先采用图像增强来扩充训练样本提高模型的鲁棒性,并添加非雪豹图像优化数据集结构;其次,使用生成式对抗网络对夜间红外灰度图像进行域迁移,融合图像迁移前后的预测结果解决夜间红外图像目标识别能力弱的问题;最后通过在主干特征提取网络中加入注意力机制细化特征图来提高RGB和红外图像中雪豹的视觉显著性,并且改进特征融合网络结构整合更多有效信息.分析对比实验和消融实验的结果表明,该方法相比Faster-RCNN、YOLOv3 和SSD的检测速度和精度更好,改进后的算法平均精确度为 97.4%且检测速度可达19 张/s,该检测模型更适合定位识别雪豹RGB与红外图像.

关键词：雪豹红外相机图像、目标检测、图像增强、域迁移、注意力机制

所属期刊栏目：28

分类号：TP391.4(计算技术、计算机技术)

资助基金：中央高校基本科研业务费专项;黑龙江省自然科学基金项目;哈尔滨市科技创新人才项

在线出版日期：2023-08-28（万方平台首次上网日期，不代表论文的发表时间）

页数：共9页

页码：108-116