Complete move of the TF conversion script

MATLAB-Community-Toolboxes-at-INCF · Sep 6, 2024 · d2e6927 · d2e6927
1 parent 9b5a3c8
commit d2e6927
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/interop/save_model_tf.py b/interop/save_model_tf.py
@@ -0,0 +1,8 @@
+#The following snippet is executed in Python 3.9 with TensorFlow 2.10. Currently, the import functionality in MATLAB support upto TF 2.10.
+#The command to install TensorFlow 2.10 is:
+#pip install tensorflow==2.10
+
+import tensorflow as tf
+model = tf.keras.models.load_model('2019_09_11_23_32_unet_single_1024_mean_absolute_error_Ai93-0450.h5')
+model.save('TF_2019_09_11_23_32_unet_single_1024_mean_absolute_error_Ai93-0450', save_format='tf')
+