如何使用 Python API 从 Google AutoML Vision 导出 tflite 模型？答案

【问题标题】：How to export a tflite model from Google AutoML Vision using Python API?如何使用 Python API 从 Google AutoML Vision 导出 tflite 模型？
【发布时间】：2021-10-31 04:04:11
【问题描述】：

我尝试从经过 Google AutoML Vision 训练的模型导出 tflite 模型。我使用 Python API 来管理数据和训练模型。不幸的是，我收到一条错误消息，告诉我“400 Unsupported model export format [tflite] for model”。即使我将模型元数据设置为

，也会发生这种情况

model_metadata = automl.ImageClassificationModelMetadata(
   train_budget_milli_node_hours = i
   model_type = "mobile-versatile-1"

并将导出 output_config 设置为

output_config = automl.ModelExportOutputConfig(
   gcs_destination=...,
   model_format="tflite"
)

当手动进入谷歌云控制台时，我奇怪地也遇到了一个问题，它在“测试和使用”选项卡中也只显示了“REST API”和“Python”选项。

因此问题是：为什么我的手动选项如此有限（Google 是否删除了导出功能）？以及如何设置我的 Python API 代码以允许 tflite 导出？谢谢！

【问题讨论】：

标签： python google-cloud-platform export google-cloud-automl

【解决方案1】：

仅当您使用“Edge”选项训练模型时，将模型导出为 TFLite / TFlite.js / Container 的选项才可用：

如果您选择“云托管”，这是默认设置，则训练后只有“REST API”和“Python”可用。

【讨论】：