Export tensorflow model to ONNX and specify variable names

Gianluca Micchi

I have a tensorflow model written through model subclassing and I want to export it to ONNX format. This is simple enough with the script attached. However, the name of the input variable to the ONNX model is args_0. How can I rename it?

import tensorflow as tf
import tf2onnx
from tensorflow.python.keras import Model
from tensorflow.python.keras.layers import Dense


class MyModel(Model):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.dense = Dense(16)

    def call(self, inputs, **kwargs):
        return self.dense(inputs)

    def to_onnx(self, output_path, opset=14):
        model_proto, _ = tf2onnx.convert.from_keras(
            self,
            input_signature=[tf.TensorSpec((1, 128))],
            opset=opset,
            output_path=output_path,
        )
        return


if __name__ == "__main__":
    output_path = "./test.onnx"
    A = MyModel()
    A.to_onnx(output_path)
kiranr

you can provide the input name in input_signature as name="input_name" so, it should look like

 input_signature=[tf.TensorSpec((1, 128), name="input_name")],

as shown in this notebook

Collected from the Internet

Please contact [email protected] to delete if infringement.

edited at
0

Comments

0 comments
Login to comment

Related

Rename variable scope of saved model in TensorFlow

Converting ONNX model to TensorFlow Lite

About names of variable scope in tensorflow

Layer names for pretrained inception v3 model (tensorflow)

Export a basic Tensorflow model to Google Cloud ML

R Loop for Variable Names to run linear regression model

What characters are allowed in tensorflow variable names?

How do I export a TensorFlow model as a .tflite file?

How to Save TensorFlow model using estimator.export_savemodel()

Run inference using Onnx model in python?

How to specify an instrumental variable model with felm()?

Withou onnx, how to convert a pytorch model into a tensorflow model manually?

How to convert Onnx model (.onnx) to Tensorflow (.pb) model

Run inference on Arm NN android with onnx model

How to convert Tensorflow 2.* trained with Keras model to .onnx format?

pytorch to Onnx(OCR model)

how to use R to export SPSS variable names and labels to *.txt file

Error on running Super Resolution Model from ONNX

Cannot export PyTorch model to ONNX

How to extract variable names and values from a model formula?

Use model variable names dynamically in Razor C#

Tensorflow: variable number of levels in a neural net model

Possibility of having variable parameter names in Django model updates

Export model to text file with out column names or headers(only values)

Trying to incorporate ML onnx model to Android App

Error in loading ONNX model with ONNXRuntime

issue while exporting torch model to onnx format

Couldn't convert pytorch model to ONNX

PyTorch normalization in onnx model