Visual_Image_Search | SHORT_DEMO_VIDEO_LINK |
Run Main_Searching_Model_Script.ipynb in a notebook environment for seeing completely error-free working demo Please refer to the Setting up the environment section for detailed setup instructions.
Currently, our model is trained mostly on women's wear products, so please use traditional wear images like saree, kurta, etc., as input.
Note: The model files are large (1.29 GB, 1.30 GB), so please use Google Drive to avoid errors.
- Download the Models folder from here.
- Upload it to your Google Drive.
- Mount your Google Drive in your notebook environment.
- Modify the paths for
prediction_by_knn_model.pkl
andproduct_maped_to_features
in your code according to your mounted Google Drive. - Install the required package: pip install ultralytics
Note: When running the cell for the first time, it may take longer. Please wait for some time.
Code to retrieve the best similar product link when uploading an image of a single product. The model performs image scan search and returns a link to a similar product from our demo data on which the model is trained. In the future, the model can be trained on millions of product images to return exact matches.
When watching Prime Video, clicking the "Shop Now" button sends the video's time frame/screenshot to this model ( written as part 2 in Main_Searching_Model_Script.ipynb file. The model give cutouts of products in video frame then user select the product cutout which he wants to search, once selected it returns a link to a similar product from our demo data on which the model is trained. In the future, the model can be trained on millions of product images to return exact matches.
We scraped data from Amazon products using amazon_scrapper.ipynb. The model utilized around 4000+ products from the following categories, each file containing approximately 100 products summed up in all_combined.csv
:
women professional dress
women party wear
women clothing
watches
traditional wear for women
stylish tops for women
saree
salwar suit
lehenga
ghagra
frock for women
formal dress for women
fashionable dresses for women
embroidery kurta
ethnic wear for women
embroidery kura
crop tops
heavy sarees
printed kurtas
red saree
bicycles
bottles
cartoys
comforters
fridges
jackets
tables
televisions
washing machines
All these data files can be found in the Scraped Data folder.