MongoDB Atlas VectorSearch Image Retrieval System

Avatar of SAVADOGO ABDOUL.
Avatar of SAVADOGO ABDOUL.

MongoDB Atlas VectorSearch Image Retrieval System

AI & Big Data Supervisor
Taipei City, Taiwan
 
The aim of this work is to find similar images, mostly similar jean, tshirt, tv or sofa image from our dataset. We leveraged MongoDB Atlas VectorSearch feature to create the image search similarity system to retrieve the information. The dataset images have been converted into embeddings and hosted on a MongoDB Atlas cluster. For retrieval, the querying image will be first converted into embeddings then via MongoDB Atlas VectorSearch function, retrieve the top k images, where k=5 in our case. Cosine similarity is used for distance calculation. Enbeddings are generated via Vision Transformer (ViT) pretrained model. https://github.com/WENDGOUNDI/VectorSearch_image_retrieval
Avatar of the user.
Please login to comment.

Published: Feb 29th 2024
23
6
0

Tools

linux
Linux
visualstudio
Visual Studio
python
Python

Similarity Search
Computer Vision
Transformers
ViT
Image Retrieval System
MongoDB
Vector Search Database
Python

Share