def test(sess, data, max_steps, display_fetches, output_dir = a.output_dir): #testing at most, process the test data once sess.run(data.iterator.initializer) max_steps = min(data.stepsPerEpoch, max_steps) filesets = [] for step in range(max_steps): try: results = sess.run(display_fetches) filesets.extend(helpers.save_images(results, output_dir, a.batch_size, a.nbTargets)) except tf.errors.OutOfRangeError: print("testing fails in OutOfRangeError") continue index_path = helpers.append_index(filesets, output_dir, a.nbTargets, a.mode) return filesets
def test(sess, data, max_steps, display_fetches, output_dir = a.output_dir): #Runs the minimum steps between what is asked by user(max_steps) and how many steps are in the full dataset (stepsPerEpoch) sess.run(data.iterator.initializer) max_steps = min(data.stepsPerEpoch, max_steps) filesets = [] for step in range(max_steps): try: #Get the results results = sess.run(display_fetches) #save the output images and add them to the list of outputed items filesets.extend(helpers.save_images(results, output_dir, a.batch_size, a.nbTargets)) except tf.errors.OutOfRangeError: print("testing fails in OutOfRangeError") continue #Create an HTML file to vizualize test results. index_path = helpers.append_index(filesets, output_dir, a.nbTargets, a.mode) return filesets