IMDEX est une bibliothèque qui permet des recherches sémantiques sur des ensembles d'images
Pour installer la bibliothèque, exécutez simplement la commande suivante sur votre
pip install imdex
Une fois la bibliothèque installée, il suffit d'importer les modules que vous allez utiliser, les modules disponibles sont répertoriés ci-dessous.
Imdex
|-Loader
|-Captioner
-Indexer
Le module de chargeur est un moyen simple d'importer et de formater toutes les images qui vont être indexées, il vous permet d'importer des images ou des dossiers uniques, il renvoie les images correctement redimensionnées et prêtes à être utilisées
from imdex.loader import load_image, load_foder
images = [load_image("./sample_img/surf.jpg")]#importing single image
images, references = load_foder("sample_img")#importing all images in a folder with their names to be used as references to the image
Ce module est responsable du sous-titrage des images, il est appelé par le module d'indexeur, mais peut être importé et utilisé indépendamment
from imdex.Captioner import Captioner
cap = Captioner()
images = []#list of images
cap.captionize(images)# returns a list of captions
L'indexeur est la classe responsable de la récupération d'images, une fois que l'image est incluse dans l'indexeur, le module de légende fournira une description en anglais qui peut être enregistrée en tant que fichier et chargé plus tard, toutes les images ont besoin d'une référence de chaîne pour les identifier, comme leurs noms par exemple. Pour interroger les images, leurs descriptions et la phrase interrogée sont converties en un vecteur d'incorporation pour mesurer la distance entre ceux-ci, lorsque les distances sont calculées, un vecteur trié est créé et renvoyé comme sortie de la fonction.
from imdex.Indexer import Indexer
idr = Indexer()
images, references = load_foder("sample_img")
idr.add_images(images, references)
idr.query("a cat is laying")
idr.save_to_csv("images.imdex")
Pour charger les descriptions, vous devez créer une nouvelle instance de la classe d'indexer et utiliser le nom du fichier comme argument dans le constructeur
idr = Indexer(captions_csv_path="images.imdex")
En tant que nouvelle bibliothèque, il a beaucoup d'améliorations à faire, et voici une liste détaillée des principales:
Tu veux aider? Envoyez une demande de fusion ou un e-mail à [email protected]