How can I use tf.string_split() in tensorflow?

Question

I want to get the extension of image files to invoke different image decoder, and I found there's a function called tf.string_split in tensorflow r0.11.

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, '.')

But when I run it, I get this error：

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

I think it may caused by the shape inference of img_src. I try to use img_src.set_shape([1,]) to fix it, but it seems not work, I get this error:

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

Also, I can't get the shape of img_src using

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

The result is img_src shape=[]. But if I use the following code:

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

The result is img_src=test_img/test1.png. Am I doing something wrong?

Try to pack img_src. Just try split_result = tf.string_split([img_src], '.') and tell me if it works — nessuno
– nessuno, Commented Oct 20, 2016 at 8:28
You're welcome. I move the comment into an answer so you can accept it. — nessuno
– nessuno, Commented Oct 21, 2016 at 5:57

nessuno · Accepted Answer · 2016-10-21 05:58:38Z

8

Just pack img_src into a tensor.

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

answered Oct 21, 2016 at 5:58

nessuno

27.1k5 gold badges87 silver badges78 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

How can I use tf.string_split() in tensorflow?

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related