版本9.02缺点

卷积去噪自动编码器简介

卷积去噪自动编码器原理其实并不复杂只是在卷积自动编码器中的输入加入了噪声。这样训练后可以用于获得破损(加入噪声)的输入和纯净的输絀之间的映射关系
上图为卷积自动编码器模型。它包含编码(Encoder)和解码(Decoder)两个部分在我们加入噪声之后,可以用于做缺陷检测之类

这裏详细的我就不介绍了,大家不懂的可以去看看别的博客

首先是util模块代码如下,这块代码是我室友写的这是真滴猛,在这里感谢我的室友Dr.Tu

当然这里的代码是通过minist的手写数据集修改的,人的训练集是Tilda纺织缺陷数据集输入大小为512*768,需要的自己去翻qiang下载吧。


文件夹路徑是这样的,其中.py文件放在autoencoder文件夹下然后开始训练!!我设了epoch为500,batchsize为1一方面是显存不大,其次改为其他的代码中也要修改输入图像均为正常图像(无缺陷)图片总共为910张,噪声因子为0.5

输入一张图,如下图所示
通过卷积去噪自动编码器结果如下,看来效果确实还不是很恏。.
两张图直接差分,再进行二值处理和膨胀腐蚀,最后缺陷结果如下图所示虽然很丑。。还很恶心

我要回帖

更多关于 探歌的9个缺点 的文章

 

随机推荐