将包含ELU的Keras模型转换为TFlite

时间:2019-05-06 14:50:56

标签: android tensorflow keras

我已经编译了keras .h5模型,该模型包含ELU激活层,并且我将它与Frozen_graph.py进行了转换,所以我有了.pb文件。

要在Android中使用.pb文件,我尝试使用“ toco” Codes that I used将其转换为TFlite文件,但出现错误:

I tensorflow/contrib/lite/toco/import_tensorflow.cc:1080] Converting unsupported operation: Elu
Here is a list of operators for which  you will need custom implementations: Elu

有什么方法可以使用ELU激活功能?

我不确定如何处理自定义实现以支持ELU。

0 个答案:

没有答案