Преобразование модели ONNX в Tensorflow Lite – pytorch_half_pixel не поддерживается ⇐ Python

[Anonymous]

Я пытаюсь преобразовать свою модель 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, так как это решило чью-то проблему с этим кодом ошибки

Подробнее здесь: https://stackoverflow.com/questions/782 ... -supported