python如何读取tfrecord文件_尝试从多个tfrecord文件读取数据

我有一个.tfrecords文件的文件夹，我想将这些文件读入网络。但是，我在一次读取多个tfrecords文件时遇到了很多麻烦。在

我的所有tfrecords文件都存储在path_to_folders列表中。我所有的功能名称都是正确的。在

我的代码如下：with tf.Session() as sess:

for folder in path_to_folders:

try:

no_grasps = int(len(os.listdir(folder)) / 5)

feature_name_images = os.path.basename(folder) + '_images'

feature_name_csv = os.path.basename(folder) + '_csv'

data_path = os.path.join(path_to_data, os.path.basename(folder) + '.tfrecords')

feature = {feature_name_images: tf.FixedLenFeature([], tf.string),

feature_name_csv: tf.FixedLenFeature([], tf.string)

}

filename_queue = tf.train.string_input_producer([data_path], num_epochs=1)

reader = tf.TFRecordReader()

_, serialized_example = reader.read(filename_queue)

features = tf.parse_single_example(serialized_example, features=feature)

image_out = tf.decode_raw(features[feature_name_images], tf.float32)

csv_out = tf.decode_raw(features[feature_name_csv], tf.float32)

image_out_reshaped = tf.reshape(image_out, [no_grasps, 200, 200, 3])

csv_out_reshaped = tf.reshape(csv_out, [no_grasps, 6])

sess.run(tf.global_variables_initializer())

sess.run(tf.local_variables_initializer())

# Create a coordinator and run all QueueRunner objects

coord = tf.train.Coordinator()

threads = tf.train.start_queue_runners(coord=coord)

# image_dataset, csv_dataset = sess.run([image_out_reshaped, csv_out_reshaped])

image_dataset = sess.run(image_out_reshaped)

coord.request_stop()

coord.join(threads)

print(image_dataset.shape, type(image_dataset))

time.sleep(2)

except tf.errors.OutOfRangeError:

print('epoch limit reached')

在读取的第一次迭代完成后(第一个tfrecords文件成功读取)，然后其余的文件告诉我已达到epoch限制，并警告：

^{pr2}$

我真的不明白为什么会这样，我想知道是否有人能帮我。在

谢谢