You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

如何在Vertex AI中获取labels.txt?导出TFLite模型缺失该文件的咨询

在Vertex AI中获取TFLite模型对应的labels.txt方法

从训练关联的存储桶或元数据提取

  • 打开Vertex AI的模型仓库,找到你的图像分类模型
  • 进入该模型的训练作业详情页,定位到训练输出的GCS存储桶路径
  • 在存储桶里找到标签映射文件(2021年的dict.txt如今可能仍用这个命名,或叫label_map.txt),下载后重命名为labels.txt即可

通过预测接口确认标签索引顺序

  • 调用模型的预测接口,上传一张已知类别的测试图,查看返回的预测结果
  • 结果里会显示每个类别对应的索引值,按照索引从小到大的顺序编写labels.txt(这是你手动创建失败的核心原因:不能按字母排序,必须匹配模型内部的索引对应关系)

检查训练/导出环节的配置

  • 如果你用AutoML图像分类训练,导出TFLite模型时,确认是否勾选了「导出标签映射文件」的选项
  • 若是自定义训练脚本,检查脚本是否在训练结束后将标签映射文件同步输出到GCS指定路径

内容的提问来源于stack exchange,提问作者Jemoy Puertollano

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.07.11 22:57:04