我试图在pix2pix.py代码的create_generator
函数中获取隐藏的图层值(并创建PNG图像)。
我添加了这些行(在create_generator
函数中):
layer_8 = tf.image.encode_png(tf.cast((tf.reshape(layers[-1], [256, 256, 3]) + 1.) * 127.5, tf.uint8))
with tf.Session() as sess:
sess.run(tf.global_variables_initializer())
png_data_ = sess.run(layer_8)
open("test_images/temp.png", 'wb').write(png_data_)
我在测试模式下运行。当我运行它时,我看到了这个警告:
2017-11-09 10:23:31.897014: I tensorflow/core/platform/cpu_feature_guard.cc:137] Your CPU supports instructions that this TensorFlow binary was not compiled to use: SSE4.1 SSE4.2 AVX AVX2 FMA
并且该程序不会继续运行。我正在使用CPU(没有GPU)。我究竟做错了什么?感谢。