def testGenerateCommandForKerasModel(self):
options = {'input_format': 'keras',
'input_path': 'tmp/model.HD5',
'quantization_bytes': 1,
'output_path': 'tmp/web_model'}
self.assertEqual(['--input_format=keras', '--quantization_bytes=1',
'tmp/model.HD5', 'tmp/web_model'],
wizard.generate_arguments(options))
def testGenerateCommandForKerasModel(self):
options = {'input_format': 'keras',
'input_path': 'tmp/model.HD5',
'weight_shard_size_bytes': '100',
'quantize_uint16': 'conv/*/weights',
'output_path': 'tmp/web_model'}
self.assertEqual(['--input_format=keras',
'--quantize_uint16=conv/*/weights',
'--weight_shard_size_bytes=100',
'tmp/model.HD5', 'tmp/web_model'],
wizard.generate_arguments(options))
def testGenerateCommandForSavedModel(self):
options = {'input_format': 'tf_saved_model',
'input_path': 'tmp/saved_model',
'saved_model_tags': 'test',
'signature_name': 'test_default',
'quantization_bytes': 2,
'skip_op_check': False,
'strip_debug_ops': True,
'output_path': 'tmp/web_model'}
self.assertEqual(['--input_format=tf_saved_model',
'--quantization_bytes=2', '--saved_model_tags=test',
'--signature_name=test_default', '--strip_debug_ops=True',
'tmp/saved_model', 'tmp/web_model'],
wizard.generate_arguments(options))
def testGenerateCommandForLayerModel(self):
options = {'input_format': 'tfjs_layers_model',
'output_format': 'keras',
'input_path': 'tmp/model.json',
'quantization_bytes': 1,
'weight_shard_size_bytes': '100',
'output_path': 'tmp/web_model'}
self.assertEqual(['--input_format=tfjs_layers_model',
'--output_format=keras',
'--quantization_bytes=1',
'--weight_shard_size_bytes=100',
'tmp/model.json',
'tmp/web_model'],
wizard.generate_arguments(options))
def testGenerateCommandForKerasSavedModel(self):
options = {'input_format': 'tf_keras_saved_model',
'output_format': 'tfjs_layers_model',
'input_path': 'tmp/saved_model',
'saved_model_tags': 'test',
'signature_name': 'test_default',
'weight_shard_size_bytes': '100',
'quantization_bytes': 1,
'skip_op_check': True,
'strip_debug_ops': False,
'output_path': 'tmp/web_model'}
self.assertEqual(['--input_format=tf_keras_saved_model',
'--output_format=tfjs_layers_model',
'--quantization_bytes=1', '--saved_model_tags=test',
'--signature_name=test_default', '--skip_op_check',
'--strip_debug_ops=False',
'--weight_shard_size_bytes=100',
'tmp/saved_model', 'tmp/web_model'],
wizard.generate_arguments(options))
def testGenerateCommandForSavedModel(self):
options = {'input_format': 'tf_saved_model',
'input_path': 'tmp/saved_model',
'saved_model_tags': 'test',
'signature_name': 'test_default',
'quantize_float16': 'conv/*/weights',
'weight_shard_size_bytes': '4194304',
'skip_op_check': False,
'strip_debug_ops': True,
'control_flow_v2': True,
'output_path': 'tmp/web_model'}
self.assertEqual(['--control_flow_v2=True',
'--input_format=tf_saved_model',
'--quantize_float16=conv/*/weights',
'--saved_model_tags=test',
'--signature_name=test_default',
'--strip_debug_ops=True',
'--weight_shard_size_bytes=4194304',
'tmp/saved_model', 'tmp/web_model'],
wizard.generate_arguments(options))