テンソルフローでtf.string_split（）を使用するにはどうすればよいですか？

Question

画像ファイルの拡張子を取得して別の画像デコーダを呼び出そうとしていて、tensorflow r0.11にtf.string_splitという関数があります。

filename_queue = tf.train.string_input_producer(filenames, shuffle=shuffle) 
reader = tf.WholeFileReader() 
img_src, img_bytes = reader.read(filename_queue) 
split_result = tf.string_split(img_src, '.') 

しかし、私はそれを実行したとき、私はこのエラーを取得する：

ValueError: Shape must be rank 1 but is rank 0 for 'StringSplit' (op: 'StringSplit') with input shapes: [], []. 

私はそれがimg_srcの形状推論によって引き起こされるかもしれないと思います。私はこのエラーを取得し、私はそれを修正するためにimg_src.set_shape([1,])を使用しようとするが、動作しないようです：

ValueError: Shapes() and (1,) are not compatible 

また、私は結果がある

tf.Print(split_result, [tf.shape(img_src)],'img_src shape=') 

を使用してimg_srcの形状を得ることができないimg_src shape=[] 。しかし、私は次のコードを使用する場合：

tf.Print(split_result, [img_src],'img_src=') 

結果はimg_src=test_img/test1.pngです。私は何か間違っているのですか？

Answer 1

img_srcをテンソルにパックしてください。

split_result = tf.string_split([img_src], '.')