imdex

imdex

IMDEX es una biblioteca que permite búsquedas semánticas sobre conjuntos de imágenes

Cómo usar

Instalación

Para instalar la biblioteca, simplemente ejecute el siguiente comando en su

 pip install imdex

Módulos

Una vez que se instala la biblioteca, solo tiene que importar los módulos que va a usar, los módulos disponibles se enumeran a continuación.

 Imdex
  |-Loader
  |-Captioner
  -Indexer

Cargador

El módulo del cargador es una forma de importar y formatear todas las imágenes que se van a indexar, le permite importar imágenes o carpetas únicas, devuelve las imágenes correctamente cambia de tamaño y listas para ser utilizadas.

 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

Subtitulante

Este módulo es responsable de subtitular las imágenes, el módulo indexador lo llama, pero puede importarse y usarse de forma independiente

 from imdex.Captioner import Captioner

cap = Captioner()

images = []#list of images

cap.captionize(images)# returns a list of captions

Indexador

El indexador es la clase responsable de recuperar imágenes, una vez que la imagen se incluye en el indexador, el módulo de subtitulante proporcionará una descripción en inglés que se puede guardar como un archivo y cargado más tarde, todas las imágenes necesitan una referencia de cadena para identificarlas, como sus nombres, por ejemplo. Para consultar las imágenes, sus descripciones y la oración consultada se convierten en un vector de incrustaciones para medir la distancia entre las que se calculan las distancias, se crea y se devuelve un vector ordenado y se devuelve como salida de la función.

 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")

Para cargar las descripciones, debe crear una nueva instancia de la clase indexador y usar el nombre del archivo como argumento en el constructor

 idr = Indexer(captions_csv_path="images.imdex")

Mejoras futuras

Como una nueva biblioteca, tiene muchas mejoras que hacer, y aquí hay una lista detallada de las principales:

• Optimización del tiempo de consulta
• Soporte de consultas de idiomas múltiples
• Mejoras en el modelo de subtítulos
• Habilitar el uso de modelos de incrustación de palabras personalizados en la búsqueda

¿Quieres ayudar? Envíe una solicitud de fusión o un correo electrónico a [email protected]