faf4bbb36d576b0c328f179eb895eb6a48f284e1,research/object_detection/builders/dataset_builder.py,,build,#Any#Any#Any#,92

Before Change

      data_map_fn = dataset.map
    dataset = data_map_fn(process_fn, num_parallel_calls=num_parallel_calls)
    if batch_size:
      dataset = dataset.apply(
          tf.contrib.data.batch_and_drop_remainder(batch_size))
    dataset = dataset.prefetch(input_reader_config.num_prefetch_batches)
    return dataset

  raise ValueError("Unsupported input_reader_config.")

After Change

      data_map_fn = dataset.map
    dataset = data_map_fn(process_fn, num_parallel_calls=num_parallel_calls)
    if batch_size:
      dataset = dataset.batch(batch_size, drop_remainder=True)
    dataset = dataset.prefetch(input_reader_config.num_prefetch_batches)
    return dataset

  raise ValueError("Unsupported input_reader_config.")

Instances