szune/SpamClassification
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix Python error due to Python version

Browse Source
This commit is contained in:
Seongjoon Lee 2022-10-19 22:39:19 +09:00
parent e096d7ecf7
commit 72e501dab8
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
python/TFLiteBufferConverter.py
View File

@ -1,13 +1,16 @@
import tensorflow as tf
tf.logging.set_verbosity( tf.logging.ERROR )
tf.compat.v1.logging.set_verbosity( tf.compat.v1.logging.ERROR )
keras_model_path = input( "Enter Keras model file path > " )
converter = tf.lite.TFLiteConverter.from_keras_model_file( keras_model_path )
keras_model = tf.keras.models.load_model(keras_model_path)
converter = tf.lite.TFLiteConverter.from_keras_model(keras_model)
#converter = tf.lite.TFLiteConverter.from_keras_model( keras_model_path )
converter.post_training_quantize = True
tflite_buffer = converter.convert()
open( 'android/model.tflite' , 'wb' ).write( tflite_buffer )
print( 'TFLite model created.')
print( 'TFLite model created.')