4580ec74800db2a9e4288d49c36c41b1c2077c42,donkeycar/parts/tflite.py,,keras_model_to_tflite,#Any#Any#Any#,3

Before Change

    verStr = tf.__version__
    if verStr.find("1.1")  == 0: // found MAJOR.MINOR match for version 1.1x.x
        converter = tf.lite.TFLiteConverter.from_keras_model_file(in_filename)
    if verStr.find("2.")  == 0: // found MAJOR.MINOR match for version 2.x.x
        new_model= tf.keras.models.load_model(in_filename) //filepath="keras_model.h5")
        converter = tf.lite.TFLiteConverter.from_keras_model(new_model)
    if data_gen is not None:
        //when we have a data_gen that is the trigger to use it to 
        //create integer weights and calibrate them. Warning: this model will
        //no longer run with the standard tflite engine. That uses only float.

After Change

def keras_model_to_tflite(in_filename, out_filename, data_gen=None):
    print("Convert model {:} to TFLite {:}".format(in_filename, out_filename))
    new_model = tf.keras.models.load_model(in_filename)
    converter = tf.lite.TFLiteConverter.from_keras_model(new_model)
    if data_gen is not None:

Instances