具有自己图像的Tensorflow自动编码器

时间:2018-08-04 10:33:47

标签: python image tensorflow autoencoder

[我是TensorFlow的新手。

我正在寻找有关自动编码以及如何使用自己的图像数据集进行训练的帮助。我想用Tensorflow编程一个自动编码器,以便能够将我的图像更好地插入其他网络,并压缩和保护它们。我仍然很难定义张量流图。我也想知道是否可以考虑使用CNN解决这个问题。

这就是我到目前为止所拥有的:(没有Tensorflow Graph等)

import tensorflow as tf

filenames = tf.constant(['test0.jpg', 'test1.jpg', 'test2.jpg', 'test3.jpg'])

safe_name = "test"

data = tf.data.Dataset.from_tensor_slices((filenames,filenames))

data = data.map(parse)
data = data.batch(2)

init = tf.global_variables_initializer()

with tf.Session() as sess:

    saver = tf.train.Saver()

    for x in range(0, 1000):
        iterator = data.make_one_shot_iterator()
        images, labels = iterator.get_next()
        sess.run(init)
        sess.run([images, labels])

    file_path = os.path.dirname(os.path.abspath(__file__)) + "/data/" + str(safe_name) + ".ckpt"
    saver.save(sess, file_path)
    print("Safe File! - Training done!")

def parse(filename):
    image_string = tf.read_file(filename)
    image_decoded = tf.image.decode_jpeg(image_string, channels=3)
    image = tf.cast(image_decoded, tf.float32)
    return image, image

如何使它成为可以正常工作的自动编码器?

0 个答案:

没有答案