Concat1D中的错误: tensors[23]的秩必须与其余的秩相同

Question

Tensorflow 1.12.0

我用我的数据集传输了SSD_Mobilenet_V1_pnp模型(与COCO一起预先训练)。

论菲顿作品--探测物体良好

转换为js。

在浏览器中运行错误：

tfjs@latest:2 Uncaught (in promise) Error: Error in concat1D: rank of tensors[23] must be the same as the rank of the rest (1)

    at assert (https://cdn.jsdelivr.net/npm/@tensorflow/tfjs@latest:2:5660)

    at https://cdn.jsdelivr.net/npm/@tensorflow/tfjs@latest:2:60833

    at Array.forEach (native)

    at assertParamsConsistent (https://cdn.jsdelivr.net/npm/@tensorflow/tfjs@latest:2:60811)

    at concat_ (https://cdn.jsdelivr.net/npm/@tensorflow/tfjs@latest:2:223560)

    at concat (https://cdn.jsdelivr.net/npm/@tensorflow/tfjs@latest:2:68004)

    at executeOp$13 (https://cdn.jsdelivr.net/npm/@tensorflow/tfjs@latest:2:800253)

    at executeOp$16 (https://cdn.jsdelivr.net/npm/@tensorflow/tfjs@latest:2:804261)

    at p (https://cdn.jsdelivr.net/npm/@tensorflow/tfjs@latest:2:811525)

    at e.processStack (https://cdn.jsdelivr.net/npm/@tensorflow/tfjs@latest:2:811935)

我做错什么了？

Answer 1

我在tfjs-node中遇到了同样的问题，这个错误可能需要您遵循https://github.com/tensorflow/tfjs-models/tree/master/coco-ssd#technical-details-for-advanced-users

注:所有的例子都在TypeScript中，但很容易翻译成JavaScript。

我详细介绍了以下三个步骤

1.删除后处理步骤

您将需要从导出的模型中删除后处理步骤。

tensorflowjs_converter --input_format=tf_saved_model \
                   --output_node_names='Postprocessor/ExpandDims_1,Postprocessor/Slice' \
                   --saved_model_tags=serve \
                   ./saved_model \
                   ./web_model

这将解决您的rank of tensors[23] must be the same as the rank of the rest (1)错误。

但是现在模型中没有任何后处理，所以需要在javascript代码中直接处理。

2.使用单类NonMaxSuppression

这样做的目的是只保留“最佳”框，并在js中进行这种过滤。

首先，您需要找到maxScores，参见示例代码这里

然后，您将需要使用NonMaxSuppression删除所有的非最高分数.

见示例代码和文档

3.在CPU后端执行NonMaxSuppression操作

您将在这里和这里中使用setBackend函数获得示例。

注意:为了释放内存，您需要在更改后端之前对张量进行tf.dispose和tf.Tensor.dispose。