python - 尝试使用tensorfow_to_barracuda.py将.pb转换为.nn时遇到问题

我收到以下错误

Converting C:\Users\qwerty\Documents\saved_model.pb to output.nn
Sorting model, may take a while.... Done!
IGNORED: GatherV2 unknown layer
Traceback (most recent call last):
  File "tensorflow_to_barracuda.py", line 26, in <module>
    tf2bc.convert(args.source_file, args.target_file, args.trim_unused_by_output, args)
  File "C:\Repos\ml-agents\ml-agents\mlagents\trainers\tensorflow_to_barracuda.py", line 1552, in convert
    i_model, args
  File "C:\Repos\ml-agents\ml-agents\mlagents\trainers\tensorflow_to_barracuda.py", line 1377, in process_model
    process_layer(n, o_context, args)
  File "C:\Repos\ml-agents\ml-agents\mlagents\trainers\tensorflow_to_barracuda.py", line 1220, in process_layer
    assert all_elements_equal(input_ranks)
AssertionError

我需要做些什么来解决这个问题？或者还有其他方法将.pb文件转换为.nn文件以在Unity Inference Engine中运行我的预训练模型

最佳答案

如果使用Tensorflow v2，则需要较旧的Tensorflow版本。 （x Tensorflow v2.卸载并安装1.15rc2有关的错误（例如）。

关于python - 尝试使用tensorfow_to_barracuda.py将.pb转换为.nn时遇到问题，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/58439392/