lucasoskorep/tensordex
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

moving to Tensorflow 2.0 and changing the test train split to use symlinks.

Browse Source
This commit is contained in:
Lucas
2022-05-18 18:37:29 -04:00
parent e1e3319653
commit 5990351a47
8 changed files with 98 additions and 83 deletions
Download Patch File Download Diff File Expand all files Collapse all files
6 - KerasToTensorflow.py
+3 -4
View File
@@ -1,11 +1,10 @@
from tensorflow.contrib.keras.api import keras
from tensorflow.contrib import lite
import tensorflow as tf
from tensorflow import keras
keras_file = "mobilenetv2.hdf5"
keras.models.load_model(keras_file)
h5_model = keras.models.load_model(keras_file)
converter = lite.TocoConverter.from_keras_model_file(keras_file)
converter = tf.lite.TFLiteConverter.from_keras_model_file(keras_file)
tflite_model = converter.convert()
with open('mobilenetv2.tflite', 'wb') as f: