213125e35ee446f4521f8959dd367d0b1a64f224,object_detection/exporter_test.py,ExportInferenceGraphTest,test_export_saved_model_and_run_inference,#ExportInferenceGraphTest#,356

Before Change

      self.assertAllClose(num_detections, [2])

  def test_export_saved_model_and_run_inference(self):
    checkpoint_path = os.path.join(self.get_temp_dir(), "model-ckpt")
    self._save_checkpoint_from_mock_model(checkpoint_path,
                                          use_moving_averages=False)
    inference_graph_path = os.path.join(self.get_temp_dir(),
                                        "saved_model")

    with mock.patch.object(
        model_builder, "build", autospec=True) as mock_builder:
      mock_builder.return_value = FakeModel(add_detection_masks=True)

After Change

      self.assertAllClose(num_detections, [2])

  def test_export_saved_model_and_run_inference(self):
    tmp_dir = self.get_temp_dir()
    trained_checkpoint_prefix = os.path.join(tmp_dir, "model.ckpt")
    self._save_checkpoint_from_mock_model(trained_checkpoint_prefix,
                                          use_moving_averages=False)
    output_directory = os.path.join(tmp_dir, "output")
    saved_model_path = os.path.join(output_directory, "saved_model")

    with mock.patch.object(
        model_builder, "build", autospec=True) as mock_builder:
      mock_builder.return_value = FakeModel(add_detection_masks=True)

Instances