Retour aux projets
IA
Featured
Image Similarity
Petit projet Python qui trouve les images les plus similaires à une image requête
Description du projet
Image Similarity (ResNet18 + Cosine Similarity)
Petit projet Python qui trouve les images les plus similaires à une image requête en utilisant :
un modèle ResNet18 pré-entraîné (TorchVision) comme extracteur de caractéristiques (embeddings)
une similarité cosinus (scikit-learn) pour comparer l’image requête à un petit dataset local
une visualisation sauvegardée (matplotlib)
Fonctionnement (résumé)
Chargement de ResNet18 pré-entraîné et suppression de la dernière couche de classification.
Pré-traitement standard ImageNet (resize/crop/normalization).
Extraction d’un vecteur de caractéristiques par image.
Calcul de la similarité cosinus entre l’image requête et toutes les images du dossier images/.
Sauvegarde d’une figure avec l’image requête + les top_k résultats dans output/.
Petit projet Python qui trouve les images les plus similaires à une image requête en utilisant :
un modèle ResNet18 pré-entraîné (TorchVision) comme extracteur de caractéristiques (embeddings)
une similarité cosinus (scikit-learn) pour comparer l’image requête à un petit dataset local
une visualisation sauvegardée (matplotlib)
Fonctionnement (résumé)
Chargement de ResNet18 pré-entraîné et suppression de la dernière couche de classification.
Pré-traitement standard ImageNet (resize/crop/normalization).
Extraction d’un vecteur de caractéristiques par image.
Calcul de la similarité cosinus entre l’image requête et toutes les images du dossier images/.
Sauvegarde d’une figure avec l’image requête + les top_k résultats dans output/.