- Mark as New
- Bookmark
- Subscribe
- Mute
- Subscribe to RSS Feed
- Permalink
- Report Inappropriate Content
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?
Link Copied
1 Reply
- Mark as New
- Bookmark
- Subscribe
- Mute
- Subscribe to RSS Feed
- Permalink
- Report Inappropriate Content
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.
Reply
Topic Options
- Subscribe to RSS Feed
- Mark Topic as New
- Mark Topic as Read
- Float this Topic for Current User
- Bookmark
- Subscribe
- Printer Friendly Page