这个是接着上一章讲解的tf.app.run()的使用来做一个实例:
现在给出一个不需要通过tf.app.run()来读取图片的代码:
import tensorflow as tf
filename='img/test.jpg'
file_constant=tf.read_file(filename)
image=tf.image.decode_jpeg(file_constant)
with tf.Session() as sess:
sess.run(tf.global_variables_initializer())
img=sess.run((image))
print (img.shape)
print (img)
把这段代码放入你的程序里面,随便下载一张jpg格式的图片,名字给位test(你也可以自己取名字,代码也要相应改掉),现在我们把这段代码改位tf.app.run()来实现:
import tensorflow as tf
tf.app.flags.DEFINE_string("image_file", "a.jpg", "define a input picture")
FLAGS = tf.app.flags.FLAGS
def main(_):
with open(FLAGS.image_file, 'rb') as img:
with tf.Session() as sess:
sess.run(tf.global_variables_initializer())
image = sess.run(tf.image.decode_jpeg(img.read()))
height = image.shape[0]
width = image.shape[1]
print (image)
print (image.shape)
if __name__=='__main__':
tf.app.run()
print ("caokaifa")