0

它的最后一层是输出为 13x13x75 矩阵的层，当它应该是 1x10 时，因为 10 是输出类的数量。

它缺少层，包括层 logits。当您下载模型时，它附带了一些 python 文件以使用它，在这些文件中您会看到必须完成的所有处理才能获得 1x10 矩阵。

关键是这段代码不应该存在，一切都必须在模型中。

.pb 扩展名的文件是什么？

有谁知道会发生什么？

4

1 回答 1

1

很抱歉造成混乱。

.pb 扩展名的文件是什么？那是您加载以进行推理的 TensorFlow 模型。

根据您收到的 13 x 13 x 75 输出猜测您创建了一个对象检测模型 (OD)，但您期望一个 1 x 10 输出向我表明您想要一个多标签模型。

多标签：返回您期望的 1x10 向量。

OD：旨在提供边界框信息以及置信度；它需要自定义代码来计算管道中不支持的；这就是为什么需要在 python 代码（'object_detection.py'）中完成最终处理的原因。

库尔特，

于 2019-05-13T18:46:00.340 回答