arXiv:2508.10509cs.CV2025-08

用分割驱动编辑生成缺陷螺栓图像,解决数据少难题。

A Segmentation-driven Editing Method for Bolt Defect Augmentation and Detection

  • 通过分割模型精准提取螺栓属性,指导缺陷编辑。
  • 生成的缺陷图像使检测准确率显著提升,超越现有方法。
  • 适合电力巡检、工业质检等需要缺陷数据增强的场景。

螺栓缺陷检测对保障输电线路安全至关重要。然而,缺陷图像稀缺和数据分布不均严重制约检测性能。为此,我们提出一种分割驱动的螺栓缺陷编辑方法(SBDE)以扩充数据集。首先,提出螺栓属性分割模型(Bolt-SAM),通过CLAHE-FFT适配器(CFA)和多部分感知掩码解码器(MAMD)增强复杂螺栓属性的分割精度,生成高质量掩码用于后续编辑。其次,设计掩码优化模块(MOD),与图像修复模型LaMa结合,构建螺栓缺陷属性编辑模型(MOD-LaMa),实现正常螺栓到缺陷螺栓的属性编辑。最后,提出编辑恢复增强(ERA)策略,将编辑后的缺陷螺栓还原至原检测场景中,扩展缺陷检测数据集。我们构建了多个螺栓数据集并进行了大量实验。结果表明,SBDE生成的缺陷图像显著优于当前主流图像编辑模型,有效提升螺栓缺陷检测性能,充分验证了该方法的有效性与应用潜力。项目代码已公开:https://github.com/Jay-xyj/SBDE。

原文摘要 · Abstract (English)

Bolt defect detection is critical to ensure the safety of transmission lines. However, the scarcity of defect images and imbalanced data distributions significantly limit detection performance. To address this problem, we propose a segmentationdriven bolt defect editing method (SBDE) to augment the dataset. First, a bolt attribute segmentation model (Bolt-SAM) is proposed, which enhances the segmentation of complex bolt attributes through the CLAHE-FFT Adapter (CFA) and Multipart- Aware Mask Decoder (MAMD), generating high-quality masks for subsequent editing tasks. Second, a mask optimization module (MOD) is designed and integrated with the image inpainting model (LaMa) to construct the bolt defect attribute editing model (MOD-LaMa), which converts normal bolts into defective ones through attribute editing. Finally, an editing recovery augmentation (ERA) strategy is proposed to recover and put the edited defect bolts back into the original inspection scenes and expand the defect detection dataset. We constructed multiple bolt datasets and conducted extensive experiments. Experimental results demonstrate that the bolt defect images generated by SBDE significantly outperform state-of-the-art image editing models, and effectively improve the performance of bolt defect detection, which fully verifies the effectiveness and application potential of the proposed method. The code of the project is available at https://github.com/Jay-xyj/SBDE.

缺陷检测数据增强图像编辑电力巡检

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。