search

bubbliiiing / ssd-keras

这是一个ssd-keras的源码，可以用于训练自己的模型。
MIT License
167 stars 68 forks source link

模型转换成tflite #16

Closed fightseed closed 3 years ago

fightseed commented 3 years ago

作者你好，请问怎样做模型转换 我使用如下代码 with CustomObjectScope({'relu6': keras.layers.ReLU(6.)}): converter = tf.lite.TFLiteConverter.from_keras_model_file('linear.h5') tflite_model = converter.convert() open("converted_model.tflite", "wb").write(tflite_model) 对模型转换成tflite，但是一致报错“ValueError: Unknown activation function:relu6”

bubbliiiing commented 3 years ago

这个。。你可以试试relu激活函数。

bubbliiiing commented 3 years ago

训练和预测