TensorFlow 中的 Azure CustomVision 认知服务导出模型缺少文件

Question

我正在使用 Azure 认知服务(又名 CustomVision 网站)来创建、训练和测试模型。据我所知，该网站的主要目标是创建可调用的 API 以在生产中运行您的模型。我应该提到我正在使用它来进行对象检测。

有时您必须支持脱机运行(这意味着您没有与 Azure 的连接等...)。我相信 Microsoft 知道并理解这一点，因为他们有一项功能可以让您以多种不同的格式导出模型(例如 TensorFlow、ONNX 等)。

我遇到的问题是，特别是当您导出到我需要的 TensorFlow 时，它只会下载卡住的模型图 (model.pb)。但是，有时您需要模型附带的 .pbtxt 文件或配置文件。我知道您可以生成 pbtxt 文件，但为此您需要 .config。

此外，导出模型后几乎没有任何有关模型的信息，例如输入图像的大小应该是多少。我希望在某个地方看到更好的记录。例如，它是 300x300 等...如果没有获取配置或 pbtxt 以及模型，您必须通过将模型加载到 TensorBoard 或类似的东西来计算出输入信息(大小、名称等) ..)。而且，我们甚至不知道模型的基线是什么，是ResNet、SSD还是等等……

那么，有人知道在导出模型时如何获取这些丢失的文件吗？或者，有人知道当您只有卡住图 .pb 文件时如何生成 pbtxt 吗？

如果没有，我会推荐这些作为 Azure 认知服务团队的改进。由于所有这些缺失的数据或信息，导出的模型确实很难使用。

谢谢!

Answer 1

许多模型架构允许更改网络输入大小，例如 Yolo，它是从 Custom Vision 导出的架构。在这种情况下，在某些地方包含固定的输入大小是没有意义的。

Netron将成为您的好 friend ，并且非常容易使用来找出有关模型的详细信息。