如何在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
相关产品推荐
相关产品推荐

