Python main Examples

Example #1

File: inspector_test.py Project: laurelwoods0102/laurelwoods-maple-market

 def test_export(self):
     # TODO(xxx): add support for tflite.
     # FLAGS.tflite_path = os.path.join(self.tempdir, 'test.tflite')
     FLAGS.saved_model_dir = os.path.join(self.tempdir, 'savedmodel')
     inspector.main(None)
     self.assertTrue(
         tf.saved_model.contains_saved_model(FLAGS.saved_model_dir))

Example #2

File: inspector_test.py Project: zihua/automl

 def test_export(self):
     FLAGS.tflite_path = os.path.join(self.tempdir, 'test.tflite')
     FLAGS.saved_model_dir = os.path.join(self.tempdir, 'savedmodel')
     inspector.main(None)
     self.assertTrue(
         tf.saved_model.contains_saved_model(FLAGS.saved_model_dir))
     self.assertTrue(tf.io.gfile.exists(FLAGS.tflite_path))

Example #3

 def test_infer(self):
   test_image = np.random.randint(0, 244, (640, 720, 3)).astype(np.uint8)
   FLAGS.input_image = os.path.join(self.tempdir, 'img.jpg')
   Image.fromarray(test_image).save(FLAGS.input_image)
   FLAGS.output_image_dir = self.tempdir
   inspector.main(None)
   self.assertTrue(tf.io.gfile.exists(os.path.join(self.tempdir, '0.jpg')))

Example #4

File: inspector_test.py Project: laurelwoods0102/laurelwoods-maple-market

 def test_benchmark(self):
     inspector.main(None)
     self.assertFalse(
         tf.io.gfile.exists(os.path.join(self.tempdir, '0.jpg')))

Example #5

File: inspector_test.py Project: laurelwoods0102/laurelwoods-maple-market

 def test_dry(self):
     FLAGS.export_ckpt = os.path.join(self.tempdir, 'model')
     inspector.main(None)
     self.assertIsNot(tf.train.get_checkpoint_state(self.tempdir), None)