tfpipeline.py 文件源码

def read_my_file_format(filename):


    record_defaults = [[""]] + [[1.0]] * 10
    components = tf.decode_csv(filename, record_defaults=record_defaults, 
        field_delim=" ")
    imgName = components[0]
    features = components[1:]
    img_contents = tf.read_file(imgName)
    img = tf.image.decode_jpeg(img_contents, channels=1)
    return img, features