python如何读取tfrecord_将数据集转化为tfrecord并读取tfrecord

**//将数据集转化为tfrecord**

import tensorflow as tf

from tensorflow.examples.tutorials.mnist import input_data

import numpy as np

def _int64_feature(value):

return tf.train.Feature(int64_list=tf.train.Int64List(value=[value]));

def _bytes_feature(value):

return tf.train.Feature(bytes_list=tf.train.BytesList(value=[value]));

mnist=input_data.read_data_sets("../mnist",dtype=tf.uint8,one_hot=True);

images=mnist.train.images;

labels=mnist.train.labels;

pixels=images.shape[1];

num_examples=mnist.train.num_examples;

filename="log/output.tfrecords";

writer=tf.python_io.TFRecordWriter(filename);

for index in range(num_examples):

image_raw=images[index].tostring();

example=tf.train.Example(features=tf.train.Features(feature={