concat1D中的错误：张量的等级[23]必须与其他等级相同

Question

Tensorflow 1.12.0

我使用数据集转换了SSD_Mobilenet_V1_pnp模型（使用COCO进行了预训练）。

在phyton上工作-很好地检测物体

转换为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。

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

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，请参见示例代码here

然后，您将需要使用 NonMaxSuppression 删除所有非最高得分。

请参见example code和documentation

3。在CPU后端执行NonMaxSuppression操作

您将使用here函数在here和setBackend中获得示例

注意：为了释放内存，您需要先tf.dispose和tf.Tensor.dispose张量才能更改后端