0e57630cea86d426a9348e8f6a9dd3a7d78b282e,research/object_detection/dataset_tools/context_rcnn/add_context_to_examples_tf1_test.py,GenerateContextDataTest,test_beam_pipeline,#GenerateContextDataTest#,328

Before Change

      sequence_key = six.ensure_binary("image/seq_id")
      max_num_elements = 10
      num_shards = 1
      pipeline = add_context_to_examples.construct_pipeline(
          input_tfrecord,
          output_tfrecord,
          sequence_key,
          max_num_elements_in_context_features=max_num_elements,
          num_shards=num_shards)
      runner.run(pipeline)
      filenames = tf.io.gfile.glob(output_tfrecord + "-?????-of-?????")
      actual_output = []
      record_iterator = tf.python_io.tf_record_iterator(path=filenames[0])

After Change

      sequence_key = six.ensure_binary("image/seq_id")
      max_num_elements = 10
      num_shards = 1
      pipeline_options = beam.options.pipeline_options.PipelineOptions(
          runner="DirectRunner")
      p = beam.Pipeline(options=pipeline_options)
      add_context_to_examples.construct_pipeline(
          p,
          input_tfrecord,
          output_tfrecord,
          sequence_key,
          max_num_elements_in_context_features=max_num_elements,
          num_shards=num_shards)
      p.run()
      filenames = tf.io.gfile.glob(output_tfrecord + "-?????-of-?????")
      actual_output = []
      record_iterator = tf.python_io.tf_record_iterator(path=filenames[0])

Instances