Downloads · 30 days
0
testing-2-kitpot
sksayril/testing-2-kitpot
testing-2-kitpot is a machine learning model from sksayril. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
This repository contains code for an image retrieval system built using ResNet50, a pre-trained convolutional neural network, and Nearest Neighbors algorithm to find similar images based on feature embeddings.
Access
Public
Updated Jan 31, 2024
Repo size
122 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pkl122 MB · 99%
At a glance
- Access
- Public
- Created
- Jan 31, 2024
- Updated
- Jan 31, 2024
- SHA
- c51db28f
Model card
From the Hugging Face model README
Image Retrieval System using ResNet50 and Nearest Neighbors
This repository contains code for an image retrieval system built using ResNet50, a pre-trained convolutional neural network, and Nearest Neighbors algorithm to find similar images based on feature embeddings.
Overview
The system leverages ResNet50, a powerful deep learning model pre-trained on ImageNet, for extracting image features. These features are stored as embeddings in a pickle file, along with associated filenames.
Files Included
app.py: Python script implementing the image retrieval system.res_vector_embeddings: Pickle file containing feature embeddings of images.filenames.pkl: Pickle file storing filenames corresponding to the image embeddings.
Getting Started
Prerequisites
- Python 3.10
- Dependencies: Keras, NumPy, scikit-learn
You can install the dependencies via:
Requirements installation
pip install -r requirements.txt
Run the Model
python app.py
- Created
- Jan 31, 2024
- Updated
- Jan 31, 2024