-
Notifications
You must be signed in to change notification settings - Fork 1
How does rubrix work?
Aashish Yadavally edited this page Aug 12, 2021
·
3 revisions
rubrix
makes use of multiple deep learning models to do several complicated image/text related tasks.
1. Image Search
- Architecture
- Video Summary
Click on the GIF below to learn how image search in rubrix
works:
2. Reverse Image Search
- Architecture
-
Summary
The reverse image search algorithm in
rubrix
can be broken down into three main steps:- Using an image recognition model, extract feature vectors for all the images in the database.
- When the user uploads an image:
- Extract all the objects in the image using YOLOv4 algorithm.
- From the inverse image index mapping different objects to the images containing them, prune to extract relevant images, i.e. those containing the objects as in the user-uploaded image.
- Calculate the similarity scores of the feature vector corresponding to the user-uploaded image with those corresponding to all pruned images, and return the top-5 results.