通信与信息技术2024,Issue(5) :19-23.

基于轮胎X光图像样本重采样图像缺陷检测

Defect detection based on resampling of tire X-ray image samples

刘韵婷 刘鑫
通信与信息技术2024,Issue(5) :19-23.

基于轮胎X光图像样本重采样图像缺陷检测

Defect detection based on resampling of tire X-ray image samples

刘韵婷 1刘鑫1
扫码查看

作者信息

  • 1. 沈阳理工大学自动化与电气工程学院,辽宁沈阳 110159
  • 折叠

摘要

针对生成对抗网络轮胎X光图像缺陷检测,训练阶段生成器会丢失部分图像特征,并且难以确定样本的潜在空间维度,导致部分不必要的图像特征重建.为了解决这些问题,构建了样本重采样生成对抗网络SRGAN(Sample Resampling Generate Adversarial Networks),生成器以VQ-VAE为基本框架,利用注意力特征融合模块(Atten-tion Feature Fusion,AFF)搭建了新的跳连层,并在SRGAN的生成器中加入了转换损失函数LVQ.最后,使用自制的轮胎X光图像数据集对SRGAN和已经提出的部分生成对抗网络模型进行训练和测试,并将得到的AUC值进行对比,进一步证明了SRGAN具有更好的图像缺陷检测能力.

Abstract

For defect detection in tire X-ray images generated by adversarial networks,during the training phase,the generator may lose some image features and find it difficult to determine the potential spatial dimensions of the samples,resulting in unnecessary image feature reconstruction.In order to solve these problems,a sample resampling generation adversarial network(SRGAN)was con-structed.The generator uses VQ-VAE as the basic framework and utilizes the Attention Feature Fusion(AFF)module to build a new hop layer.The conversion loss function LVQ was added to the SRGAN generator.Finally,the self-made tire X-ray image dataset was used to train and test SRGAN and the proposed partially generated adversarial network model,and the obtained AUC values were com-pared,further proving that SRGAN has better image defect detection ability.

关键词

生成对抗网络/VQ-VAE/AFF/轮胎X光图像缺陷检测

Key words

Generative adversarial network/VQ-VAE/AFF/Tire X-ray image defect detection

引用本文复制引用

出版年

2024
通信与信息技术
四川省通信学会

通信与信息技术

影响因子:0.223
ISSN:1672-0164
段落导航相关论文