基于并行混合注意力与多尺度频域损失的织物瑕疵图像生成
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

国家重点研发计划(2024YFB4614200); 国家自然科学基金(52575602); 浙江省“尖兵领雁”研发攻关计划(2025C01088)


Fabric Defect Image Generation Based on Parallel Hybrid Attention and Multi-scale Frequency Domain Loss
Author:
Affiliation:

Fund Project:

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

    织物瑕疵检测模型的性能高度依赖充足且多样的高质量样本, 而实际采集成本高昂、类别分布失衡, 严重制约了模型的泛化能力. 为此, 本文提出一种基于并行混合注意力与多尺度频域损失的织物瑕疵图像生成方法. 首先, 在局部生成器中引入并行混合注意力机制, 通过自适应融合空间门注意力与通道注意力, 增强了瑕疵定位的准确性与关键特征的表达. 其次, 提出多尺度频域损失, 在频率域内对幅值和相位进行约束, 协同感知损失共同还原织物的高频纹理. 最后, 在判别器中嵌入自注意力以捕捉长距离依赖, 增强图像的全局结构一致性. 在自建数据集上的实验结果表明, 本文方法生成图像的SSIMPSNR以及LPIPS值分别达到了0.964、32.797 dB和0.085, 较现有方法在感知质量与细节保真度上更具优势. 通过对比数据增强前后的检测性能, YOLOv5、YOLOv8、DEIM和D-FINE的平均精度分别提高了1.4%、1.2%、1.1%和0.8%.

    Abstract:

    The performance of fabric defect detection models heavily relies on a sufficient quantity of diverse and high-quality samples. However, data collection is costly, and class distributions are often imbalanced, which severely restricts the model’s generalization ability. To address this issue, this study proposes a fabric defect image generation method based on parallel hybrid attention and multi-scale frequency-domain loss. First, a parallel hybrid attention mechanism is introduced into the local generator, which adaptively fuses spatial gating attention and channel attention to enhance defect localization accuracy and improve the representation of key features. Second, a multi-scale frequency-domain loss is proposed to constrain both amplitude and phase, and it is combined with perceptual loss to better reconstruct high-frequency fabric textures. Finally, self-attention is embedded into the discriminator to capture long-range dependencies and enhance the global structural consistency of the generated images. Experimental results on a self-built dataset show that the proposed method achieves SSIM, PSNR, and LPIPS values of 0.964, 32.797 dB, and 0.085, respectively, outperforming existing methods in perceptual quality and detail fidelity. By comparing detection performance before and after data augmentation, the average precision of YOLOv5, YOLOv8, DEIM, and D-FINE improves by 1.4%, 1.2%, 1.1%, and 0.8%, respectively.

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

王思怡,向忠,陈燕兵,胡旭东.基于并行混合注意力与多尺度频域损失的织物瑕疵图像生成.计算机系统应用,,():1-13

复制
分享
相关视频

文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2026-01-30
  • 最后修改日期:2026-03-26
  • 录用日期:
  • 在线发布日期: 2026-08-21
  • 出版日期:
文章二维码
您是第位访问者
版权所有:中国科学院软件研究所 京ICP备05046678号-3
地址:北京市海淀区中关村南四街4号,邮政编码:100190
电话:010-62661041 传真: Email:csa@iscas.ac.cn
技术支持:北京勤云科技发展有限公司

京公网安备 11040202500063号