Turicreate iOS图像相似度

Question

我一直在关注Turicreate教程链接here

我能够按照说明成功训练模型。我也可以借助那里提供的代码在iOS应用程序中使用该模型。但我无法弄清楚如何根据此模型返回的距离获得最相似的实际图像。此外，当我在iPhone上运行它时，它返回我无法解释的偏移量和元素。请看截图。

我在iOS应用程序中的目标是输入图像，将输入图像传递给模型，然后显示实际输出5或10个最相似的图像，而不仅仅是距离。

Answer 1

请确保您具有id列，或已使用以下方法将id列添加到SFrame：

reference_data = reference_data.add_row_number()

这是一个代码段，用于创建image_similarity模型并为示例图像获取十个最相似的图像。我使用五十作为k的值，以演示如何从相似度图中的较大组中抽取最相似的n。

similarity_graph = model.similarity_graph(k=50)
similar_images = similarity_graph.edges
# pick a sample image from reference_data
sample_index = 3
sample_similar_images = similar_images[similar_images["__src_id"]==sample_index].sort("rank")
# get 10 most similar images
most_similar_ids = sample_similar_images["__dst_id"][0:10]
most_similar_images = reference_data.filter_by(most_similar_ids, "id")

有关从CoreML模型解释距离MultiArray的详细信息，请参见Turi Create documentation中的示例代码。元素值对应于距离。