By: Mike Palei
This post will showcase the use of cosine similarity algorithm in Neo4J and also provide examples in addition to the available documentation.
Without further ado, here is the problem set:
I had a collection of people’s images and I wanted to find images showing the same person. Since the collection was not labeled, I could not tackle this task as a pure classificaton problem.
I manually created a list of persons of interest and downloaded their “prototypical” pictures from the web.
Next step was to extract embeddings from both the “prototypical” images and from each image in my collection. For this I used a pretrained Resnet50 model from keras.applications.