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))