4f14cb629462c9e685ea92a961a80a8c5dcda83a,object_detection/meta_architectures/ssd_meta_arch_test.py,SsdMetaArchTest,test_predict_results_have_correct_keys_and_shapes,#SsdMetaArchTest#,119

Before Change

    init_op = tf.global_variables_initializer()
    with self.test_session() as sess:
      sess.run(init_op)
      prediction_out = sess.run(prediction_dict)
      self.assertAllEqual(prediction_out["box_encodings"].shape,
                          expected_box_encodings_shape_out)
      self.assertAllEqual(
          prediction_out["class_predictions_with_background"].shape,

After Change

      with self.test_session(graph=tf_graph) as sess:
        sess.run(init_op)
        prediction_out = sess.run(prediction_dict,
                                  feed_dict={
                                      preprocessed_input_placeholder:
                                      np.random.uniform(
                                          size=(batch_size, 2, 2, 3))})
      self.assertAllEqual(prediction_out["box_encodings"].shape,
                          expected_box_encodings_shape_out)
      self.assertAllEqual(

Instances