Skip to content

GitLab

Unverified Commit ea6d6aab authored by pkulzc's avatar pkulzc Committed by GitHub
Browse files

Merge pull request #4205 from vladpaunescu/master 

Fixed Bug regarding tfrecord shuffling in object_detection
parents 9f58547c e04280eb
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
research/object_detection/utils/dataset_util.py
View file @ ea6d6aab
...@@ -134,7 +134,7 @@ def read_dataset(file_read_func, decode_func, input_files, config): ...@@ -134,7 +134,7 @@ def read_dataset(file_read_func, decode_func, input_files, config):
file_read_func, cycle_length=config.num_readers, file_read_func, cycle_length=config.num_readers,
block_length=config.read_block_length, sloppy=config.shuffle)) block_length=config.read_block_length, sloppy=config.shuffle))
if 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( tensor_dataset = records_dataset.map(
decode_func, num_parallel_calls=config.num_parallel_map_calls) decode_func, num_parallel_calls=config.num_parallel_map_calls)
return tensor_dataset.prefetch(config.prefetch_size) return tensor_dataset.prefetch(config.prefetch_size)
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