Merge pull request tensorflow#4205 from vladpaunescu/master

pkulzc · web-flow · commit ea6d6aabe5c1 · 2018-05-14T21:00:42.000-07:00
Fixed Bug regarding tfrecord shuffling in object_detection
diff --git a/research/object_detection/utils/dataset_util.py b/research/object_detection/utils/dataset_util.py
@@ -134,7 +134,7 @@ def read_dataset(file_read_func, decode_func, input_files, config):
           file_read_func, cycle_length=config.num_readers,
           block_length=config.read_block_length, sloppy=config.shuffle))
   if config.shuffle:
-    records_dataset.shuffle(config.shuffle_buffer_size)
+    records_dataset = records_dataset.shuffle(config.shuffle_buffer_size)
   tensor_dataset = records_dataset.map(
       decode_func, num_parallel_calls=config.num_parallel_map_calls)
   return tensor_dataset.prefetch(config.prefetch_size)
