feed_input.py 文件源码

def readfile(filename):
    try:
        reader = tf.WholeFileReader()
        key,value = reader.read(filename)
        image = tf.image.decode_jpeg(value, channels=3)
        image = tf.image.resize_images(image, 224, 224)
        float_image = tf.div(tf.cast(image,tf.float32), 255)
        return float_image
    except:
        print -1
        return readfile(filename)