如何在对象检测API中使用GCP视觉训练的模型

时间:2019-11-06 13:30:14

标签: python-3.x tensorflow google-cloud-platform object-detection-api

我在GCP中通过视觉训练了对象检测模型,如何在google GitHub提供的普通tensorflow对象检测api中使用该模型? 它提供了3种导出模型的选项,其中哪一种以及如何使用?

1 个答案:

答案 0 :(得分:0)

您需要将模型从云保存到本地驱动器中,并将此模型加载到对象检测api中,函数load_model通常从URL下载经过预先​​训练的模型,您需要在此处提供本地保存的模型的路径。您正在寻找的API是tf.saved_model.load,还可以正确更新PATH_TO_LABELS中标签的路径。