论文解析:区域和强度可控的GAN用于工业图像缺陷生成,数据扩充,分割_区域和强度可控的gan用于工业图像缺陷生成论文下载-程序员宅基地

技术标签: 生成对抗网络  机器学习  计算机视觉  图像处理  深度学习  

前言

论文名称:
Region- and Strength-Controllable GAN for Defect Generation and Segmentation in Industrial Images
论文地址:
IEEE TII Region- and Strength-Controllable GAN for Defect Generation and Segmentation in Industrial Images

摘要

计算机视觉的深度学习 (DL) 基于海量、多样化和注释良好的训练集取得了显着的成果。 然而,很难收集涵盖所有可能特征的缺陷数据集,尤其是对于小而弱的缺陷。 因此,提出了一种缺陷区域和强度可控的缺陷图像生成方法。 被视为使用生成对抗网络 (GAN) 的图像修复,生成的缺陷区域是通过使用缺陷掩码来控制的。 此外,基于缺陷和非缺陷之间的特征连续性在潜在变量空间中构造缺陷方向向量来控制缺陷强度,从而实现缺陷掩模和图像之间的一对多对应。此外,还设计了缺陷注意力损失以强制生成模型关注缺陷区域。 实验上,我们的方法生成的图像质量和多样性更好,从而显着提高了缺陷分割性能(Kolektor 表面缺陷(KSD)和金属钩缺陷(MHD)上的联合交集(IoU)分别为 63.20% 和 61.86%) 数据集),尤其是对于小而弱的缺陷。

1. 解决问题

由于近年来深度学习 (DL) 在计算机视觉性能方面取得了显着进步,将 DL 应用于表面缺陷分割已成为许多工业系统的研究热点,例如金属、纺织品和太阳能电池。 DL 在计算机视觉应用中取得如此显着成果的主要原因之一是海量、多样化和注释良好的训练集的可用性。但是,由于生产线的良率较高,很难采集到大量的缺陷图像。因此,考虑到缺陷数据集的大小和多样性,很难满足构建令人满意的基于 DL 的缺陷分割模型的要求。具体来说,在工业现场采集的典型缺陷图像往往具有明显的特征,称为正常缺陷。对应于正常样本,边界缺陷样本是基于工业检测标准位于检测超曲面附近的缺陷样本;边界缺陷样本难以收集且具有较高的值密度。由于尺寸和强度是工业检测最重要的指标,区域小、特征弱的缺陷是常见的边界缺陷样本,如图1所示。 很难收集到可以覆盖所有可能特征的缺陷,包括正常和边界缺陷样本,以满足DL检测模型的要求。此外,为在生产现场收集的各种缺陷图像构建像素级注释是一项耗时且费力的任务。
在这里插入图片描述
图 1. 正常和边界缺陷。 检测超曲面附近的边界缺陷很难收集。

因此提出一种基于GAN的区域和强度可控的生成方法,生成样本,尤其是边界样本

2 方法

在这里插入图片描述
图2.1训练
在这里插入图片描述
图2.2 测试

整体过程可以看做为缺陷部分遮盖的图像复原并在隐变量空间中控制缺陷特征的强弱

2.1 缺陷掩码输入模块

训练过程中,基于掩码将缺陷区域遮挡,形成缺陷区域被遮挡的缺陷图像
在这里插入图片描述
图3.1 掩码输入-训练

测试过程中将基于工业检测标准等先验知识,构建缺陷掩码,并将其覆盖在无缺陷图像中,以控制生成缺陷图像的位置,尺寸,形状。在这里插入图片描述
图3.2 掩码输入-测试

2.2 缺陷方向向量模块

基于隐变量特征变换控制缺陷强度

缺陷方向向量构建阶段

在这里插入图片描述
图4.1 缺陷方向向量构建子模块-训练

基于缺陷图像和无缺陷图像,构建逻辑回归,并以逻辑回归的参数 w,作为缺陷方向向量 Vd

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

隐变量生成阶段

在这里插入图片描述
将输入的隐变量,求取其到分类边界的距离,然后在该变量与分类面之间进行差值,获取新样本

2.3 损失模块

损失模块的主要思路是对分别对整张图像和缺陷局部构建损失,包括重建损失和对抗损失。

3 实验

本文实验是在KSDD数据集上构建

其对弱对比度缺陷的检测能力显著提高

整体检测性能在定量上也显著提升

版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/u014031149/article/details/121520638

智能推荐

攻防世界_难度8_happy_puzzle_攻防世界困难模式攻略图文-程序员宅基地

文章浏览阅读645次。这个肯定是末尾的IDAT了,因为IDAT必须要满了才会开始一下个IDAT,这个明显就是末尾的IDAT了。,对应下面的create_head()代码。,对应下面的create_tail()代码。不要考虑爆破,我已经试了一下,太多情况了。题目来源:UNCTF。_攻防世界困难模式攻略图文

达梦数据库的导出(备份)、导入_达梦数据库导入导出-程序员宅基地

文章浏览阅读2.9k次,点赞3次,收藏10次。偶尔会用到,记录、分享。1. 数据库导出1.1 切换到dmdba用户su - dmdba1.2 进入达梦数据库安装路径的bin目录,执行导库操作  导出语句:./dexp cwy_init/[email protected]:5236 file=cwy_init.dmp log=cwy_init_exp.log 注释:   cwy_init/init_123..._达梦数据库导入导出

js引入kindeditor富文本编辑器的使用_kindeditor.js-程序员宅基地

文章浏览阅读1.9k次。1. 在官网上下载KindEditor文件,可以删掉不需要要到的jsp,asp,asp.net和php文件夹。接着把文件夹放到项目文件目录下。2. 修改html文件,在页面引入js文件:<script type="text/javascript" src="./kindeditor/kindeditor-all.js"></script><script type="text/javascript" src="./kindeditor/lang/zh-CN.js"_kindeditor.js

STM32学习过程记录11——基于STM32G431CBU6硬件SPI+DMA的高效WS2812B控制方法-程序员宅基地

文章浏览阅读2.3k次,点赞6次,收藏14次。SPI的详情简介不必赘述。假设我们通过SPI发送0xAA,我们的数据线就会变为10101010,通过修改不同的内容,即可修改SPI中0和1的持续时间。比如0xF0即为前半周期为高电平,后半周期为低电平的状态。在SPI的通信模式中,CPHA配置会影响该实验,下图展示了不同采样位置的SPI时序图[1]。CPOL = 0,CPHA = 1:CLK空闲状态 = 低电平,数据在下降沿采样,并在上升沿移出CPOL = 0,CPHA = 0:CLK空闲状态 = 低电平,数据在上升沿采样,并在下降沿移出。_stm32g431cbu6

计算机网络-数据链路层_接收方收到链路层数据后,使用crc检验后,余数为0,说明链路层的传输时可靠传输-程序员宅基地

文章浏览阅读1.2k次,点赞2次,收藏8次。数据链路层习题自测问题1.数据链路(即逻辑链路)与链路(即物理链路)有何区别?“电路接通了”与”数据链路接通了”的区别何在?2.数据链路层中的链路控制包括哪些功能?试讨论数据链路层做成可靠的链路层有哪些优点和缺点。3.网络适配器的作用是什么?网络适配器工作在哪一层?4.数据链路层的三个基本问题(帧定界、透明传输和差错检测)为什么都必须加以解决?5.如果在数据链路层不进行帧定界,会发生什么问题?6.PPP协议的主要特点是什么?为什么PPP不使用帧的编号?PPP适用于什么情况?为什么PPP协议不_接收方收到链路层数据后,使用crc检验后,余数为0,说明链路层的传输时可靠传输

软件测试工程师移民加拿大_无证移民,未受过软件工程师的教育(第1部分)-程序员宅基地

文章浏览阅读587次。软件测试工程师移民加拿大 无证移民,未受过软件工程师的教育(第1部分) (Undocumented Immigrant With No Education to Software Engineer(Part 1))Before I start, I want you to please bear with me on the way I write, I have very little gen...

随便推点

Thinkpad X250 secure boot failed 启动失败问题解决_安装完系统提示secureboot failure-程序员宅基地

文章浏览阅读304次。Thinkpad X250笔记本电脑,装的是FreeBSD,进入BIOS修改虚拟化配置(其后可能是误设置了安全开机),保存退出后系统无法启动,显示:secure boot failed ,把自己惊出一身冷汗,因为这台笔记本刚好还没开始做备份.....根据错误提示,到bios里面去找相关配置,在Security里面找到了Secure Boot选项,发现果然被设置为Enabled,将其修改为Disabled ,再开机,终于正常启动了。_安装完系统提示secureboot failure

C++如何做字符串分割(5种方法)_c++ 字符串分割-程序员宅基地

文章浏览阅读10w+次,点赞93次,收藏352次。1、用strtok函数进行字符串分割原型: char *strtok(char *str, const char *delim);功能:分解字符串为一组字符串。参数说明:str为要分解的字符串,delim为分隔符字符串。返回值:从str开头开始的一个个被分割的串。当没有被分割的串时则返回NULL。其它:strtok函数线程不安全,可以使用strtok_r替代。示例://借助strtok实现split#include <string.h>#include <stdio.h&_c++ 字符串分割

2013第四届蓝桥杯 C/C++本科A组 真题答案解析_2013年第四届c a组蓝桥杯省赛真题解答-程序员宅基地

文章浏览阅读2.3k次。1 .高斯日记 大数学家高斯有个好习惯:无论如何都要记日记。他的日记有个与众不同的地方,他从不注明年月日,而是用一个整数代替,比如:4210后来人们知道,那个整数就是日期,它表示那一天是高斯出生后的第几天。这或许也是个好习惯,它时时刻刻提醒着主人:日子又过去一天,还有多少时光可以用于浪费呢?高斯出生于:1777年4月30日。在高斯发现的一个重要定理的日记_2013年第四届c a组蓝桥杯省赛真题解答

基于供需算法优化的核极限学习机(KELM)分类算法-程序员宅基地

文章浏览阅读851次,点赞17次,收藏22次。摘要:本文利用供需算法对核极限学习机(KELM)进行优化,并用于分类。

metasploitable2渗透测试_metasploitable2怎么进入-程序员宅基地

文章浏览阅读1.1k次。一、系统弱密码登录1、在kali上执行命令行telnet 192.168.26.1292、Login和password都输入msfadmin3、登录成功,进入系统4、测试如下:二、MySQL弱密码登录:1、在kali上执行mysql –h 192.168.26.129 –u root2、登录成功,进入MySQL系统3、测试效果:三、PostgreSQL弱密码登录1、在Kali上执行psql -h 192.168.26.129 –U post..._metasploitable2怎么进入

Python学习之路:从入门到精通的指南_python人工智能开发从入门到精通pdf-程序员宅基地

文章浏览阅读257次。本文将为初学者提供Python学习的详细指南,从Python的历史、基础语法和数据类型到面向对象编程、模块和库的使用。通过本文,您将能够掌握Python编程的核心概念,为今后的编程学习和实践打下坚实基础。_python人工智能开发从入门到精通pdf

推荐文章

热门文章

相关标签