Почему я получаю сообщение об ошибке при использовании tf2onnx для преобразования модели keras в onnx?

У меня есть модель CNN, которую я обучил с помощью TensorFlow.Keras (v2.10) и сохранил лучшие контрольные точки в виде файла .h5. Я хотел преобразовать его в onnx, используя следующую команду:

      python -m tf2onnx.convert --saved-model weights-chkpt-07-0.65.h5 --output AMR_CNN.onnx

Но я получил эту ошибку, которую я не совсем понимаю:

      2022-11-08 11:07:15,257 - WARNING - '--tag' not specified for saved_model. Using --tag serve
Traceback (most recent call last):
  File "/usr/lib/python3.8/runpy.py", line 194, in _run_module_as_main
    return _run_code(code, main_globals, None,
  File "/usr/lib/python3.8/runpy.py", line 87, in _run_code
    exec(code, run_globals)
  File "/usr/local/lib/python3.8/dist-packages/tf2onnx/convert.py", line 706, in <module>
    main()
  File "/usr/local/lib/python3.8/dist-packages/tf2onnx/convert.py", line 238, in main
    graph_def, inputs, outputs, initialized_tables, tensors_to_rename = tf_loader.from_saved_model(
  File "/usr/local/lib/python3.8/dist-packages/tf2onnx/tf_loader.py", line 611, in from_saved_model
    _from_saved_model_v2(model_path, input_names, output_names,
  File "/usr/local/lib/python3.8/dist-packages/tf2onnx/tf_loader.py", line 549, in _from_saved_model_v2
    imported = tf.saved_model.load(model_path, tags=tag)  # pylint: disable=no-value-for-parameter
  File "/usr/local/lib/python3.8/dist-packages/tensorflow/python/saved_model/load.py", line 800, in load
    result = load_partial(export_dir, None, tags, options)["root"]
  File "/usr/local/lib/python3.8/dist-packages/tensorflow/python/saved_model/load.py", line 905, in load_partial
    loader_impl.parse_saved_model_with_debug_info(export_dir))
  File "/usr/local/lib/python3.8/dist-packages/tensorflow/python/saved_model/loader_impl.py", line 57, in parse_saved_model_with_debug_info
    saved_model = parse_saved_model(export_dir)
  File "/usr/local/lib/python3.8/dist-packages/tensorflow/python/saved_model/loader_impl.py", line 115, in parse_saved_model
    raise IOError(
OSError: SavedModel file does not exist at: weights-chkpt-07-0.65.h5/{saved_model.pbtxt|saved_model.pb}

Как я могу решить эту проблему? и чем это вызвано.

0 ответов

Другие вопросы по тегам