image_inputs.py 文件源码

def read_my_file_format(filename):

    record_defaults = [[""]] + [[0]]

    components = tf.decode_csv(filename, record_defaults=record_defaults, field_delim=" ")
    imgName = components[0]
    label = components[1:]
    img_contents = tf.read_file(imgName)
    img = tf.image.decode_jpeg(img_contents, channels=3)

    return img, label