tf2onnx_keras_model_conversion_with_input_signature.py

1import tensorflow as tf
2import tf2onnx
3import onnx
4
5# 1. Create or load a simple Keras model
6model = tf.keras.Sequential([
7    tf.keras.layers.Dense(10, input_shape=(5,), name="input"),
8    tf.keras.layers.Activation('relu', name="output")
9])
10
11# 2. Define the input signature (required for conversion)
12# Shape is [batch_size, input_dimension]
13spec = (tf.TensorSpec((None, 5), tf.float32, name="input"),)
14
15# 3. Convert the model to ONNX
16output_path = "model.onnx"
17model_proto, _ = tf2onnx.convert.from_keras(model, input_signature=spec, opset=13)
18
19# 4. Save the ONNX model to a file
20with open(output_path, "wb") as f:
21    f.write(model_proto.SerializeToString())
22
23print(f"Model successfully converted and saved to {output_path}")