convert.py 文件源码

def read_raw_images(sess, data_set):
    filename = ['./data/' + data_set + '_data.bin']
    filename_queue = tf.train.string_input_producer(filename)
    print filename
    record_bytes = (FLAGS.height) * (FLAGS.width) * FLAGS.depth + 1
    image_bytes = (FLAGS.height) * (FLAGS.width) * FLAGS.depth
    reader = tf.FixedLengthRecordReader(record_bytes=record_bytes)
    key, value = reader.read(filename_queue)
    record_bytes = tf.decode_raw(value, tf.uint8)
    #record_label = tf.decode_raw(value, tf.int32)
    tf.train.start_queue_runners(sess=sess)
    for i in range(0, 10):
        result = sess.run(record_bytes)
        print i, result[0], len(result)
        image = result[1:len(result)]
        print image