0

我正在使用以下命令将tensorflow模型转换为表单tensorflowjs

tensorflowjs_converter 
    --input_format=tf_saved_model 
    --output_node_names="my_output_node" \
    --saved_model_tags=serve my_saved_model_dir \
    ./web_model

我遇到了以下神秘错误：

ValueError: Unsupported Ops in the model before optimization NonMaxSuppression, ResizeArea

我的图表中确实有这些操作。我需要把它们换成更tensorflowjs友好的东西吗？

我深入谷歌，只发现了对以下标志的引用，我可以添加到tensorflowjs_converter 命令--skip_op_check=SKIP_OP_CHECK \ 这确实编译了，但是在尝试为js模型提供服务时，我遇到了类似于上面的 js 错误：

错误：不支持 Tensorflow Op：ResizeArea

任何想法如何修改我的图表或我的命令来导航这个？

谢谢

4

1 回答 1

2

简短的回答是肯定的，您需要更改它们。TensorflowJS 将更改操作以进行优化，但并非所有操作都具有等效的 TFJS 版本。支持的操作的完整列表在这里：https ://github.com/tensorflow/tfjs-converter/blob/master/docs/supported_ops.md

奇怪的是，“NonMaxSuppression”似乎确实在列表中，但 ResizeArea 不在，并且 100% 不起作用。

另一种方法是自己创建一个自定义操作，并使用该代码，但我不确定如何在 TFJS 中执行此操作。

于 2019-02-27T16:05:56.660 回答