将 ONNX 模型转换为 Tensorflow Lite - 不支持 pytorch_half_pixel

Question

我正在尝试将 ONNX 模型转换为 Tensorflow Lite 格式。简单的代码但出现此错误。我更新了我的 onnx 版本，但没有运气

import onnx
import tensorflow as tf
import onnx_tf
#
#
# README: This file converts an onnx model to tflite
#
#
#

onnx_model_path = '/home/sfrye/segmentation/segmentation_checkpoints/efficientnet/modified-new.onnx'

onnx_model = onnx.load(onnx_model_path)

tf_model = onnx_tf.backend.prepare(onnx_model)
tf_model.export_graph("tflite_model.tf")

这是错误

RuntimeError: in user code:

    File "/home/sfrye/miniconda3/envs/mars_env/lib/python3.8/site-packages/onnx_tf/backend_tf_module.py", line 99, in __call__  *
        output_ops = self.backend._onnx_node_to_tensorflow_op(onnx_node,
    File "/home/sfrye/miniconda3/envs/mars_env/lib/python3.8/site-packages/onnx_tf/backend.py", line 347, in _onnx_node_to_tensorflow_op  *
        return handler.handle(node, tensor_dict=tensor_dict, strict=strict)
    File "/home/sfrye/miniconda3/envs/mars_env/lib/python3.8/site-packages/onnx_tf/handlers/handler.py", line 58, in handle  *
        cls.args_check(node, **kwargs)
    File "/home/sfrye/miniconda3/envs/mars_env/lib/python3.8/site-packages/onnx_tf/handlers/backend/resize.py", line 125, in args_check  *
        exception.OP_UNSUPPORTED_EXCEPT(
    File "/home/sfrye/miniconda3/envs/mars_env/lib/python3.8/site-packages/onnx_tf/common/exception.py", line 50, in __call__  *
        raise self._func(self.get_message(op, framework))

    RuntimeError: Resize coordinate_transformation_mode=pytorch_half_pixel is not supported in Tensorflow.

我尝试更新我的onnx，因为这修复了某人的错误代码问题

Answer 1

不完全相同的场景，但我过去通过在我的

interpolation=bilinear

层中显式设置

Upsampling2D

解决了类似的问题。检查您拥有的任何调整大小图层。 但这需要重新训练模型。