Intel® Distribution of OpenVINO™ Toolkit
Community assistance about the Intel® Distribution of OpenVINO™ toolkit, OpenCV, and all aspects of computer vision-related on Intel® platforms.
6403 Discussions

When using the TF data api and iterators, how do you name the input?

idata
Employee
575 Views

Here is how I input data into my network:

 

iterator = TFRecordParser.input_fn(batch_size, record_path)

 

sess.run(iterator.initializer)

 

image, ground_truth = iterator.get_next()

 

prediction = network(image)

 

So "image" is the data going into my network, and "prediction" is the output. How do I name these so that I can compile with mvNCCompile?

0 Kudos
1 Reply
idata
Employee
318 Views

Nevermind, I figured it out. Basically, when you go to make your inference only python file, you have to delete the iterator code anyways because you don't need it. So just replace

 

image, ground_truth = iterator.get_next()

 

with

 

image = tf.placeholder(tf.float32, [None, 480, 640, 3], name="input")

 

even if you didn't use the feed_dict pattern.

0 Kudos
Reply