import tensorflow.contrib.layers as lays import numpy as np from skimage import transform import tensorflow as tf from tensorflow.examples.tutorials.mnist import input_data import matplotlib.pyplot as plt #定义具有编码器和解码器的网络 def autoencoder(inputs): net = lays.conv2d(inputs,32,[5,5],stride=2,padding='SAME') net = lays.conv2d(net,16,[5,5],stride=2,padding='SAME') net = lays.conv2d(net,8,[5,5],stride=4,padding='SAME') # 解码器 net = lays.conv2d_transpose(net,16,[5,5],stride=4,padding='SAME') net = lays.conv2d_transpose(net,32,[5,5],stride=2,padding='SAME') net = lays.conv2d_transpose(net,1,[5,5],stride=2,padding='SAME',activation_fn=tf.nn.tanh) return net def resize_batch(imgs): #该函数将图像调整为32×32,这样维数可以减少一半 imgs = imgs.reshape((-1,28,28,1)) resized_imgs =np.zeros((imgs.shape[0],32,32,1)) for i in range(img
去噪自编码器tensorflow实现
最新推荐文章于 2022-08-31 13:31:55 发布