Turicreate iOS Image Схожесть

Question

Я перехожу по ссылке учебника Turicreate здесь

Я могу успешно тренировать модель в соответствии с инструкциями. Я также могу использовать модель в приложении для iOS с помощью приведенного там кода. Но я не могу понять, как я могу получить наиболее похожие реальные изображения, основываясь на расстояниях, которые возвращает эта модель. Кроме того, когда я запускаю его на iPhone, он возвращает мне смещение и элемент, которые я не могу интерпретировать. Пожалуйста, смотрите скриншот.

Моя цель в приложении iOS состоит в том, чтобы ввести изображение, передать это входное изображение в модель, а затем показать фактический вывод 5 или 10 наиболее похожих изображений, а не только расстояния.

Answer 1

Будьте уверены, что у вас есть id столбец или добавили id колонка к вашему фрейму, используя:

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")

Подробнее об интерпретации расстояния MultiArray от модели CoreML см. В примере кода в документации Turi Create. Значение элемента соответствует расстоянию.