示例#1
0
 def testValidateInputPathForKerasSavedModel(self):
   self.assertNotEqual(True, wizard.validate_input_path(
       self._tmp_dir, 'keras_saved_model'))
   self._create_keras_saved_model()
   save_dir = os.path.join(self._tmp_dir, SAVED_MODEL_DIR)
   self.assertEqual(True, wizard.validate_input_path(
       save_dir, 'keras_saved_model'))
示例#2
0
 def testValidateInputPathForKerasModel(self):
   self.assertNotEqual(True,
                       wizard.validate_input_path(self._tmp_dir, 'keras'))
   self._create_hd5_file()
   save_dir = os.path.join(self._tmp_dir, HD5_FILE_NAME)
   self.assertEqual(True, wizard.validate_input_path(
       save_dir, 'keras'))
示例#3
0
 def testValidateInputPathForTFHub(self):
     self.assertNotEqual(
         True, wizard.validate_input_path(self._tmp_dir, 'tf_hub'))
     self.assertEqual(
         True,
         wizard.validate_input_path("https://tfhub.dev/mobilenet",
                                    'tf_hub'))
示例#4
0
  def testValidateInputPathForLayersModel(self):
    self.assertNotEqual(True,
                        wizard.validate_input_path(self._tmp_dir, 'keras'))
    self._create_layers_model()
    save_dir = os.path.join(self._tmp_dir)
    self.assertEqual(True, wizard.validate_input_path(
        save_dir, 'tfjs_layers_model'))

    save_dir = os.path.join(self._tmp_dir, 'model.json')
    self.assertEqual(True, wizard.validate_input_path(
        save_dir, 'tfjs_layers_model'))