[我是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
如何使它成为可以正常工作的自动编码器?