首页 > 解决方案 > 从 Azure 自定义视觉导出以使用 TensorFlow 的模型不完整

问题描述

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

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

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

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

有谁知道会发生什么?

标签: azuretensorflowartificial-intelligenceazure-cognitive-servicesmicrosoft-custom-vision

解决方案


很抱歉造成混乱。

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

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

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

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

库尔特,


推荐阅读