TF-Lite非最大值抑制

Question

我正在尝试使用tf.image.non_max_suppression或tf.image.combined_non_max_suppression转换图形，但这两个应用程序接口调用都会产生类似“flex既不是自定义op也不是tf.CombinedNonMaxSuppression op”的错误。我的设置是TF2.3.1，python 3.7，Windows 10。

我知道一些tf函数不支持转换到TF-Lite，但下面的链接显示了用于非最大抑制的tfl函数。

https://tensorflow.google.cn/mlir/tfl_ops#tflnon_max_suppression_v4_tflnonmaxsuppressionv4op

为了使用tfl.non_max_suppression_vx函数，我需要做什么才能在我的函数上运行转换器？

Answer 1

TF Lite不支持非max抑制操作。如果你想使用它们，你必须通过TF lib来使用，当转换时，添加这些行

converter.experimental_new_converter=True

converter.target_spec.supported_ops = [tf.lite.OpsSet.TFLITE_BUILTINS,
                                   tf.lite.OpsSet.SELECT_TF_OPS]

还有另一种方法，您应该通过TF Lite或类似的工具中支持的操作来重写NMS op。如果你重写成功了，请告诉我。谢谢。