如何将滤镜转换回RGB值?

时间:2019-05-26 18:09:40

标签: tensorflow conv-neural-network

我正在阅读有关卷积神经网络的教程,并且陷入了第一步。当我运行plt.imshow(output[0])时,我被告知尺寸错误。我知道最后一个列表的大小必须为3或0才能使imshow正常工作,但是由于过滤器的数量,当前的列表大小必须为64。

我尝试将过滤器的数量设置为3,这出乎意料地起作用,但是我不认为那是我应该做的...

import matplotlib.pyplot as plt
import sklearn.datasets as skd


image = skd.load_sample_image("china.jpg")
input_layer = tf.placeholder(tf.float32, [None, 427, 640, 3])
conv_1 = tf.layers.conv2d(input_layer, filters=64, kernel_size=[3, 3],padding="same", activation=tf.nn.relu)
init = tf.global_variables_initializer()
with tf.Session() as sess:
    sess.run(init)
    output = sess.run(conv_1, feed_dict={input_layer: [image]})
plt.imshow(output[0])
plt.show()

0 个答案:

没有答案