2017-10-15 7 views
0

Tensorflowを使用して簡単なCNNを構築しようとしています。問題は、単純な.pngファイルをCNNに送ることができないということです。pngファイルを読み込み、cnnを送ります。Tensorflow

>>> filename = tf.constant("training/a1.png") 
>>> filename 
<tf.Tensor 'Const_1:0' shape=() dtype=string> 
>>> image_string = tf.read_file(filename) 
>>> image_string 
<tf.Tensor 'ReadFile_1:0' shape=() dtype=string> 
>>> image_decoded = tf.image.decode_png(image_string) 


>>> image_decoded 
<tf.Tensor 'DecodePng_1:0' shape=(?, ?, ?) dtype=uint8> 

上記のコードのとおりです。 tf.image.decode_png(image_string)は、未知のテンソルの形状を返します。

ありがとうございましたvladimir-bystricky!わかった。他の人に役立つかもしれないコードはここにあります。

>>> import tensorflow as tf 
>>> filename = tf.constant("training/a1.png") 
>>> image_string = tf.read_file(filename) 
>>> image_decoded = tf.image.decode_png(image_string) 
>>> shape = tf.shape(image_decoded) 
>>> sess = tf.Session() 
>>> print(sess.run(shape)) 
[360 360 4] 

答えて

1

それである右の行動、あなたのコードであなただけのグラフを作成し、TFは本当にこのステップのPNGファイルを読み込むが、唯一あなたがグラフを起動すると、それを行います操作を作成していないため、セッション。操作の結果は、未知の形状(この瞬間)のTensorです。あらかじめ定義されたサイズでサイズ変更またはトリミングすることができます。

関連する問題