task.py 文件源码

def read_image_from_disk(input_queue):
    label = input_queue[1]
    file_name = tf.reduce_join([TENSOR_BASE_PATH, input_queue[0]], 0)
    file_contents = tf.read_file(file_name)
    rgb_image = tf.image.decode_jpeg(file_contents, channels=FLAGS.image_channels,
        name="decode_jpeg")
    rgb_image  = tf.image.convert_image_dtype(rgb_image, dtype=tf.float32,
        name="image_convert_float32")
    rgb_image = tf.image.resize_images(rgb_image,
        [FLAGS.image_size, FLAGS.image_size])

    return rgb_image, label