CIFAR10 二进制数据读取

 

1 分析

• 构造文件队列
• 读取二进制数据并进行解码
• 处理图片数据形状以及数据类型，批处理返回
• 开启会话线程运行

2 代码

• 定义CIFAR类，设置图片相关的属性

class CifarRead(object):
    """
    二进制文件的读取，tfrecords存储读取
    """

    def __init__(self):
        # 定义一些图片的属性
        self.height = 32
        self.width = 32
        self.channel = 3

        self.label_bytes = 1
        self.image_bytes = self.height * self.width * self.channel
        self.bytes = self.label_bytes + self.image_bytes

• 实现读取数据方法bytes_read(self, file_list)

  • 构造文件队列

  # 1、构造文件队列
  file_queue = tf.train.string_input_producer(file_list)
  

  • tf.FixedLengthRecordReader(bytes)读取

  # 2、使用tf.FixedLengthRecordReader(bytes)读取
  # 默认必须