在讀取和打印TFRecord格式數(shù)據(jù)時(shí)，必須加入線(xiàn)程機(jī)制，否則打印不出任何結(jié)果

import tensorflow as tffilename_queue = tf.train.string_input_producer(['train.tfrecords'], shuffle=True) reader = tf.TFRecordReader() _, serialized_example = reader.read(filename_queue) features = tf.parse_single_example(serialized_example,features={'image_raw': tf.FixedLenFeature([], tf.string),'label': tf.FixedLenFeature([], tf.int64),'height': tf.FixedLenFeature([], tf.int64),'width': tf.FixedLenFeature([], tf.int64),'depth': tf.FixedLenFeature([], tf.int64)})image = tf.decode_raw(features['image_raw'], tf.float64) image = tf.reshape(image, [28, 28]) image.set_shape([28, 28]) image = tf.cast(image, tf.float32) label = tf.cast(features['label'], tf.int32)with tf.Session() as sess:init_op = tf.global_variables_initializer()sess.run(init_op)coord = tf.train.Coordinator()threads = tf.train.start_queue_runners(coord=coord)for i in range(10):example, l = sess.run([image, label])print(example)coord.request_stop()coord.join(threads)

?

總結(jié)

以上是生活随笔為你收集整理的读取打印TFRecord格式数据的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。