Commit b18d79de authored by Amir MOHAMMADI's avatar Amir MOHAMMADI
Browse files

Do not create the dataset in the script

parent 8325c6f5
......@@ -75,9 +75,6 @@ from bob.learn.tensorflow.utils.commandline import \
from bob.core.log import setup, set_verbosity_level
logger = setup(__name__)
# tensorflow 1.4 and above.
Dataset = tf.data.Dataset
def bio_generator(database, preprocessor, groups, number_of_parallel_jobs,
biofile_to_label, multiple_samples=False):
......@@ -169,10 +166,8 @@ def main(argv=None):
database, preprocessor, groups, number_of_parallel_jobs,
biofile_to_label, multiple_samples)
dataset = Dataset.from_generator(
generator, output_types, output_shapes)
predict_input_fn = bio_predict_input_fn(dataset)
predict_input_fn = bio_predict_input_fn(generator,
output_types, output_shapes)
predictions = estimator.predict(
predict_input_fn,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment