如何从 TensorFlow 项目中创建端点

Question

张量流的新手。我发现了这个很酷的项目，你可以上传一张鸟的照片，它会识别鸟类的种类。在这里能找到它：

https://tfhub.dev/google/aiy/vision/classifier/birds_V1/1

任何想法我可以如何从中创建一个端点，以便我可以将照片发布到并接收鸟类类型作为响应

Answer 1

您可以使用 tensorflow 进行预训练

# Import pre-trained bird classification model from Tensorflow Hub 
URL = 'https://tfhub.dev/google/aiy/vision/classifier/birds_V1/1'           
bird = hub.KerasLayer(URL, input_shape=(224,224,3))
model=tf.keras.Sequential([bird])
model.summary()

获得模型后，您可以将图像传递给您想要预测的模型。在将图像传递给模型之前，您必须将图像转换为 numpy 数组。

bird=np.array(bird)/255.0

然后你可以使用

result=np.argmax(model.predict(bird[np.newaxis,...]))

完整代码请参考gist。谢谢你。