tflite 没有找到内置操作码 'STRIDED_SLICE' 版本 '6' 的操作

Question

嗨，我正在尝试将 tflite 上传到 android 工作室，但是
未找到内置操作码 'STRIDED_SLICE' 版本“6” 的操作
出来。这是将 .h5 文件转换为 tflite 的 python 代码。

fastspeech_concrete_function = fastspeech2.inference_tflite.get_concrete_function()
converter = tf.lite.TFLiteConverter.from_concrete_functions(
[fastspeech_concrete_function]
)
converter.optimizations = [tf.lite.Optimize.DEFAULT]
converter.target_spec.supported_ops = [tf.lite.OpsSet.TFLITE_BUILTINS,
                                   tf.lite.OpsSet.SELECT_TF_OPS,
                                   ]
 
tflite_model = converter.convert()

with open('fastspeech.tflite', 'wb') as f:
    f.write(tflite_model)

我添加了

tf.strided_slice, tf.raw_ops.StridedSlice in
converter.target_spec.supported_ops

但没用我能怎么做？
我正在使用如下所示的 tflite 版本

implementation 'org.tensorflow:tensorflow-lite:0.0.0-nightly'
implementation 'org.tensorflow:tensorflow-lite-select-tf-ops:0.0.0-nightly'
implementation 'org.tensorflow:tensorflow-lite-support:0.0.0-nightly'

我试过 tensorflow==2.2.0 & 2.3.0 & 2.4.1

Answer 1

implementation 'org.tensorflow:tensorflow-lite:2.5.0'
implementation 'org.tensorflow:tensorflow-lite-select-tf-ops:0.0.0-nightly-SNAPSHOT'
implementation 'org.tensorflow:tensorflow-lite-support:0.0.0-nightly'

我用这些解决了问题。但我不确定上面几行是否会解决其他问题。

Answer 2

我也有这个错误。这是因为我通过 exporter_main_v2.py 文件冻结模型，但通过 export_tflite_graph_tf2.py 是必要的。进一步转换为tflite格式

tflite 没有找到内置操作码 'STRIDED_SLICE' 版本 '6' 的操作

tflite Didn't find op for builtin opcode 'STRIDED_SLICE' version '6'

tensorflow

tensorflow-lite