()
| 32 | |
| 33 | |
| 34 | def test_image_generator(): |
| 35 | # Generates an iterator over images |
| 36 | with tf.compat.v1.Session() as sess: |
| 37 | input_data = tf.compat.v1.data.make_one_shot_iterator(dataset.test( |
| 38 | flags.data_dir)).get_next() |
| 39 | try: |
| 40 | while True: |
| 41 | yield sess.run(input_data) |
| 42 | except tf.errors.OutOfRangeError: |
| 43 | pass |
| 44 | |
| 45 | |
| 46 | def run_eval(interpreter, input_image): |