论文解析：区域和强度可控的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数据集上构建

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

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

本文链接：https://blog.csdn.net/u014031149/article/details/121520638

原作者删帖不实内容删帖广告或垃圾文章投诉

智能推荐

c# 调用c++ lib静态库_c#调用lib-程序员宅基地

文章浏览阅读2w次，点赞7次，收藏51次。四个步骤1.创建C++ Win32项目动态库dll 2.在Win32项目动态库中添加外部依赖项 lib头文件和lib库3.导出C接口4.c#调用c++动态库开始你的表演...①创建一个空白的解决方案，在解决方案中添加 Visual C++ , Win32 项目空白解决方案的创建：添加Visual C++ , Win32 项目这......_c#调用lib

deepin/ubuntu安装苹方字体-程序员宅基地

文章浏览阅读4.6k次。苹方字体是苹果系统上的黑体，挺好看的。注重颜值的网站都会使用，例如知乎：font-family: -apple-system, BlinkMacSystemFont, Helvetica Neue, PingFang SC, Microsoft YaHei, Source Han Sans SC, Noto Sans CJK SC, W..._ubuntu pingfang

html表单常见操作汇总_html表单的处理程序有那些-程序员宅基地

文章浏览阅读159次。表单表单概述表单标签表单域按钮控件demo表单标签表单标签基本语法结构<form action="处理数据程序的url地址“ method=”get|post“ name="表单名称”></form><!--method将表单中的数据传送给服务器处理，get方式直接显示在url地址中，数据可以被缓存，且长度有限制；而post方式数据隐藏传输，_html表单的处理程序有那些

PHP设置谷歌验证器（Google Authenticator）实现操作二步验证_php otp 验证器-程序员宅基地

文章浏览阅读1.2k次。使用说明:开启Google的登陆二步验证（即Google Authenticator服务）后用户登陆时需要输入额外由手机客户端生成的一次性密码。实现Google Authenticator功能需要服务器端和客户端的支持。服务器端负责密钥的生成、验证一次性密码是否正确。客户端记录密钥后生成一次性密码。下载谷歌验证类库文件放到项目合适位置(我这边放在项目Vender下面)https://github.com/PHPGangsta/GoogleAuthenticatorPHP代码示例://引入谷_php otp 验证器

【Python】matplotlib.plot画图横坐标混乱及间隔处理_matplotlib更改横轴间距-程序员宅基地

文章浏览阅读4.3k次，点赞5次，收藏11次。matplotlib.plot画图横坐标混乱及间隔处理_matplotlib更改横轴间距

docker — 容器存储_docker 保存容器-程序员宅基地

文章浏览阅读2.2k次。①Storage driver 处理各镜像层及容器层的处理细节，实现了多层数据的堆叠，为用户提供了多层数据合并后的统一视图②所有 Storage driver 都使用可堆叠图像层和写时复制（CoW）策略③docker info 命令可查看当系统上的 storage driver主要用于测试目的，不建议用于生成环境。_docker 保存容器

随便推点

网络拓扑结构_网络拓扑csdn-程序员宅基地

文章浏览阅读834次，点赞27次，收藏13次。网络拓扑结构是指计算机网络中各组件（如计算机、服务器、打印机、路由器、交换机等设备）及其连接线路在物理布局或逻辑构型上的排列形式。这种布局不仅描述了设备间的实际物理连接方式，也决定了数据在网络中流动的路径和方式。不同的网络拓扑结构影响着网络的性能、可靠性、可扩展性及管理维护的难易程度。_网络拓扑csdn

JS重写Date函数，兼容IOS系统_date.prototype 将所有 ios-程序员宅基地

文章浏览阅读1.8k次，点赞5次，收藏8次。IOS系统Date的坑要创建一个指定时间的new Date对象时，通常的做法是：new Date("2020-09-21 11:11:00")这行代码在 PC 端和安卓端都是正常的，而在 iOS 端则会提示 Invalid Date 无效日期。在IOS年月日中间的横岗许换成斜杠，也就是new Date("2020/09/21 11:11:00")通常为了兼容IOS的这个坑，需要做一些额外的特殊处理，笔者在开发的时候经常会忘了兼容IOS系统。所以就想试着重写Date函数，一劳永逸，避免每次ne_date.prototype 将所有 ios