Move from keras-onnx to tf2onnx

The former is incompatible with Tensorflow 2.4+ [1] [1] onnx/keras-onnx#651
EMCLab-Sinica · Jul 1, 2021 · ab22c96 · ab22c96
1 parent 2cc57ab
commit ab22c96
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/data/squeezenet_cifar10.onnx b/data/squeezenet_cifar10.onnx
diff --git a/utils/export_squeezenet_cifar10.py b/utils/export_squeezenet_cifar10.py
@@ -3,13 +3,13 @@
 
 import tensorflow as tf
 import onnx
-import keras2onnx
+import tf2onnx.convert
 
 with open(os.path.join(sys.argv[1], 'models', 'squeeze_net.json')) as f:
     model_json = f.read()
 model = tf.keras.models.model_from_json(model_json)
 model.load_weights(os.path.join(sys.argv[1], 'models', 'squeeze_net.h5'))
 
-onnx_model = keras2onnx.convert_keras(model, model.name)
+onnx_model, _ = tf2onnx.convert.from_keras(model)
 
 onnx.save(onnx_model, 'squeezenet_cifar10.onnx')