TensorFlow实现卷积自编码器对图像进行去噪

一、概念介绍

自编码器是一种执行数据压缩的网络架构，其中的压缩和解压缩功能是从数据本身学习得到的，而非人为手工设计的。自编码器的两个核心部分是编码器和解码器，它将输入数据压缩到一个潜在表示空间里面，然后再根据这个表示空间将数据进行重构得到最后的输出数据。编码器和解码器都是用神经网络构建的，整个网络的构建方式和普通的神经网络类似，通过最小化输入和输出之间的差异来得到最好的网络。

二、作用

1.    图像去噪；

2.    数据压缩降维。

    但是它的图像压缩性能不如JPEG、MP3等传统压缩方法，并且自编码器泛化到其他数据集方面有困难。

三、卷积自编码器实现：

    1.    加载数据：

            我们的数据基于MNIST数据集，首先需要下载数据并且放在MNIST_data目录下，可以从文章后面提供的链接下载，也可以自行找网上的资源进行下载。目录结构：

，

MNIST数据集：

加载数据集：

%matplotlib inline   import numpy as np import tensorflow as tf import matplotlib.pyplot as plt   from tensorflow.examples.tutorials.mnist import input_data mnist = input_data.read_data_sets('MNIST_data/', validation_size=0)

    2.    数据可视化：

            查看一张图片：

img = mnist.train.images[2] plt.imshow(img.reshape((28, 28)), cmap='Greys_r')

            输出：

        

    3.    构建神经网络结构：

            网络的编码器部分将是一个典型的卷积金字塔。每一个卷积层后面都有一个最大池化层来减少维度。解码器需要从一个窄的表示转换成一个宽的重构图像。例如，表示可以是4x4x8 的最大池化层。这是编码器的输出，也是译码器的输入。我们想要从解码器中得到一个28x28x1图像，所以我们需要从狭窄的解码器输入层返回。这是网络的示意图：

            

            这里我们最后的编码器层有大小4x4x8=128。原始图像的大小为28x28x1=784，因此编码的矢量大约是原始图像大小的16%。这些只是每个层的建议大小。网络的深度和大小都可以更改，但请记住，我们的目标是找到输入数据的一个小表示。

            在编码阶段，我们使用卷积层和最大池化层来不断减小输入的维度，在解码器阶段，需要使用反卷积将4x4x8的图片还原到原来的28x28x1。我们使用的这种反卷积方法叫做去卷积，关于反卷积的知识，可以查看这篇文章。在Tensorflow中，很容易使用