imdex

IMDEX

IMDEX é uma biblioteca que permite pesquisas semânticas sobre conjuntos de imagens

Como usar

Instalação

Para instalar a biblioteca, basta executar o seguinte comando em seu

 pip install imdex

Módulos

Depois que a biblioteca estiver instalada, basta importar os módulos que você usará, os módulos disponíveis estão listados abaixo.

 Imdex
  |-Loader
  |-Captioner
  -Indexer

Carregador

O módulo carregador é uma maneira de importar e formatar todas as imagens que serão indexadas, permite importar imagens ou pastas únicas, ele retorna as imagens corretamente redimensionar e prontas para serem usadas

 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

Legendador

Este módulo é responsável por legendar as imagens, é chamado pelo módulo indexador, mas pode ser importado e usado de forma independente

 from imdex.Captioner import Captioner

cap = Captioner()

images = []#list of images

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

Indexador

O indexador é a classe responsável pela recuperação de imagens, uma vez que a imagem é incluída no indexador, o módulo do legendador fornecerá uma descrição em inglês que pode ser salva como um arquivo e carregado posteriormente, todas as imagens precisam de uma referência de string para identificá -las, como seus nomes, por exemplo. Para consultar as imagens, suas descrições e a frase consultada são convertidas em um vetor de incorporação para medir a distância entre elas, quando as distâncias são calculadas, um vetor classificado é criado e retornado como saída da função.

 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 carregar as descrições, você deve criar uma nova instância da classe Indexer e usar o nome do arquivo como um argumento no construtor

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

Melhorias futuras

Como uma nova biblioteca, ele tem muitas melhorias a serem feitas e aqui está uma lista detalhada das principais:

• Otimização do tempo de consulta
• Suporte de consultas com vários idiomas
• Melhorias no modelo de legenda
• Habilite o uso de modelos de incorporação de palavras personalizadas na pesquisa

Quer ajudar? Envie uma solicitação de mesclagem ou um e -mail para [email protected]