将 tensorflow saved_model 转换为 tensorflowjs 的问题

Question

我想将经过训练的 python 模型 (.pb) 转换为 tensorflowjs 模型。为此，我首先使用 estimator.export_savedmodel 函数保存了模型，然后在 Google Colab 上运行 tensorflowjs_converter 命令。但是，没有为 tensorflowjs 创建文件。转换也给出了很多警告并以错误结束。

这是完整的代码，请运行以查看完整的输出： https://colab.research.google.com/drive/19k2s8eHpQY9Trps9dyaxPp0HqHWp5qpb

问题的原因是什么，我该如何解决？

部分输出：

Instructions for updating:
Use `tf.compat.v1.graph_util.extract_sub_graph`
Traceback (most recent call last):
  File "/usr/local/bin/tensorflowjs_converter", line 8, in <module>
    sys.exit(pip_main())
  File "/usr/local/lib/python3.6/dist-packages/tensorflowjs/converters/converter.py", line 638, in pip_main
    main([' '.join(sys.argv[1:])])
  File "/usr/local/lib/python3.6/dist-packages/tensorflowjs/converters/converter.py", line 642, in main
    convert(argv[0].split(' '))
  File "/usr/local/lib/python3.6/dist-packages/tensorflowjs/converters/converter.py", line 591, in convert
    strip_debug_ops=args.strip_debug_ops)
  File "/usr/local/lib/python3.6/dist-packages/tensorflowjs/converters/tf_saved_model_conversion_v2.py", line 435, in convert_tf_saved_model
    strip_debug_ops=strip_debug_ops)
  File "/usr/local/lib/python3.6/dist-packages/tensorflowjs/converters/tf_saved_model_conversion_v2.py", line 141, in optimize_graph
    ', '.join(unsupported))
ValueError: Unsupported Ops in the model before optimization
ParallelDynamicStitch, StringSplit, Unique, RegexReplace, DynamicPartition, StringToHashBucketFast, ParseExample, LookupTableFindV2, LookupTableSizeV2, SparseFillEmptyRows, StringJoin, AsString, SparseSegmentSqrtN, HashTableV2

编辑： 好像不支持： https://github.com/tensorflow/tfjs/issues/2322

Answer 1

这是因为您的模型包含 tensorflow.js 尚不支持的操作。似乎您错过了粘贴的输出中缺少的操作名称。请随时更新缺少操作名称的输出或在tensorflow.js repo 中提交功能请求以提供更多详细信息。