This was a small project that I decided to try out with the help of Image retrival and Image Classification using CNNs.
For a given image, this model will search up the relevant similar images in the dataset.
To use this code in Google Colab:
- Replace the drive mount with the path of the colab folder present in your drive
- Replace the image dataset folder paths and image paths to your dataset
To use this code locally:
- Download the code
- Replace the image dataset folder paths and image paths to your dataset