基于图像翻译的红外弱小目标数据增强算法
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

上海科技馆科研专项项目(No.SSTM/SOPZY03R2KZ20230100001);中国科技国防计划项目(No.2021JCJQJJ0041)资助。


Infrared small dim target data augmentation algorithm based on image translation
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    为了缓解在红外弱小目标检测问题中图像数据稀缺的问题,提出了一种基于图像翻译的红外弱小目标图像数据增强算法。该方法是一个两阶段的图像生成算法,首先引入额外的可见光图像,通过U GAT IT模型学习可见光和红外图像之间的映射,将可见光图像转化为红外背景图像。为了解决图像翻译过程中的过拟合问题,提出了通道正则化方法,使红外和可见光图像的通道信息量保持一致。接着,设计了一个基于视觉Transformer结构的自编码器,学习红外小目标的分布特征,以遮挡重构的方式在得到的红外背景图像上合成弱小目标。本方法在SIATD数据集上进行训练和测试,实验结果表明提出的数据增强方法在三个模型上使检测指标得到了一定提升,其中在YOLOv3模型上AP指标提高了137,证明了提出的数据增强算法的有效性,能够提高目标检测模型在红外弱小目标检测任务中的表现。

    Abstract:

    In order to alleviate the scarcity of image datain infrared small dimtarget detection,an image augmentation algorithm based on image to image translation is proposed.This method is a two stage image generation algorithm.First,additional visible image is introduced,and the mapping between visible and infrared image is learned by U GAT IT model,converting the visible image into infrared background image.In order to solve the problem of overfitting in image translation,a channel regularization method is proposed to make the channel information of infrared and visible images consistent Then,an auto encoder based on vision Transformer structure is designed to learn the distribution characteristics of infrared small targets and synthesize small targets on the obtained infrared background images in the way of masking and reconstructing.The method is trained and tested on SIATD data sets.The experimental results show that the proposed data augmentation algorithm can improve the detection indexes on three models to a certain extent,among which the AP index of the YOLOv3 model increases by 1.37%,which proves the effectiveness of the proposed data augmentation algorithm and can improve the performance of the target detection model in the infrared small dim target detection task.

    参考文献
    相似文献
    引证文献
引用本文

廖彦彬,季钰翔,傅志凌,杨海,王喆.基于图像翻译的红外弱小目标数据增强算法[J].激光与红外,2025,55(2):304~312
LIAO Yan-bin, JI Yu-xiang, FU Zhi-ling, YANG Hai, WANG Zhe. Infrared small dim target data augmentation algorithm based on image translation[J]. LASER & INFRARED,2025,55(2):304~312

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2025-02-24
  • 出版日期:
×
最新公告
根据主办单位安排,编辑部2025年春节放假时间为1月26日~2月9日,2月10日起正常上班。放假期间投稿系统正常运行,其他业务暂缓办理。
    考虑寒假和快递物流等影响,为避免信件丢失,2025年第一、二期的杂志样刊等相关信件拟在3月份前后通过邮政快递发出,可通过本刊平台浏览、下载当期封面、目录、文章电子版。
    另外,本刊电子书架现已上线,可点击平台首页“电子书架”或扫码在线阅读。