From a36660b4e7c0ddb51c0aa5ad4b2a9aa6cf483349 Mon Sep 17 00:00:00 2001 From: "Steinar H. Gunderson" Date: Fri, 9 Feb 2018 20:05:25 +0000 Subject: [PATCH] Enable prefetching of images. --- voxel_flow_train.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/voxel_flow_train.py b/voxel_flow_train.py index 7d39968..736255b 100755 --- a/voxel_flow_train.py +++ b/voxel_flow_train.py @@ -52,14 +52,17 @@ def train(dataset_frame1, dataset_frame2, dataset_frame3): data_list_frame1 = dataset_frame1.read_data_list_file() dataset_frame1 = tf.data.Dataset.from_tensor_slices(tf.constant(data_list_frame1)) dataset_frame1 = dataset_frame1.repeat().shuffle(buffer_size=1000000, seed=1).map(_read_image) + dataset_frame1 = dataset_frame1.prefetch(100) data_list_frame2 = dataset_frame2.read_data_list_file() dataset_frame2 = tf.data.Dataset.from_tensor_slices(tf.constant(data_list_frame2)) dataset_frame2 = dataset_frame2.repeat().shuffle(buffer_size=1000000, seed=1).map(_read_image) + dataset_frame2 = dataset_frame2.prefetch(100) data_list_frame3 = dataset_frame3.read_data_list_file() dataset_frame3 = tf.data.Dataset.from_tensor_slices(tf.constant(data_list_frame3)) dataset_frame3 = dataset_frame3.repeat().shuffle(buffer_size=1000000, seed=1).map(_read_image) + dataset_frame3 = dataset_frame3.prefetch(100) batch_frame1 = dataset_frame1.batch(FLAGS.batch_size).make_initializable_iterator() batch_frame2 = dataset_frame2.batch(FLAGS.batch_size).make_initializable_iterator() -- 2.39.2