Required Python >= 3.8
Required Packages
- pytorch
- sklearn
- tkinter
- numpy
- matplotlib
- pillow
- warnings
Download dataset here. This data set consists of 16,000 images of apparels including some outliers.
efficient-net.ipynb has the required code to generate embeddings from images. Generate required embeddings from the dataset which gives a .npy file with all embedings. Some pretrained embedings can be found here.
Use the recommender.ipynb to train and test the embeddings data.