def _tfrecord_to_graph_ops(self, num_epochs):
with tf.variable_scope('tfrec_to_graph'):
file_queue = tf.train.string_input_producer(
self.filenames_list,
name=self.name+'_file_queue',
num_epochs=num_epochs
)
reader = tf.TFRecordReader(
options=tf.python_io.TFRecordOptions(
compression_type=self.compression
), name=self.name+'_tfrec_reader'
)
_, tfrecord = reader.read(file_queue)
features, targets = parse_mnist_tfrec(
tfrecord, self.name, self.features_shape
)
return features, targets
评论列表
文章目录