tf_helper.py

def read_label(queue):
  reader = tf.TextLineReader(1)
  _, buffer = reader.read(queue)
  return decode_label(buffer)
    
def decode_label(buffer):
  return tf.decode_csv(buffer, [[''],[0]])

def read_image(queue):
  reader = tf.WholeFileReader()
  key, buffer = reader.read(queue)
  return key, decode_image(buffer)
    
def decode_image(buffer):
  return tf.image.decode_jpeg(buffer, 3)