imdex

Имдекс

Imdex - это библиотека, которая позволяет семантическое поиск по наборам изображений

Как использовать

Установка

Чтобы установить библиотеку, просто запустите следующую команду на вашем

 pip install imdex

Модули

После установки библиотеки вам просто нужно импортировать модули, которые собираетесь использовать, доступные модули перечислены ниже.

 Imdex
  |-Loader
  |-Captioner
  -Indexer

Погрузчик

Модуль загрузчика - это способ импорта и форматирования всех изображений, которые будут индексированы, он позволяет импортировать отдельные изображения или папки, он возвращает изображения правильно изменение размера и готово к использованию

 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

Подпись

Этот модуль отвечает за подписание изображений, его называют модулем индексатора, но он может быть импортирован и используется независимо

 from imdex.Captioner import Captioner

cap = Captioner()

images = []#list of images

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

Индексер

Индексер является классом, отвечающим за получение изображений, как только изображение включено в индексатор, модуль подписи предоставит описание на английском языке, которое можно сохранить в виде файла, а позже всем изображениям нужна ссылка на строку, чтобы идентифицировать их, как их имена, например. Чтобы запросить изображения, их описания и запрашиваемое предложение преобразуются в вектор встраиваний для измерения расстояния между ними, когда расстояния рассчитываются, отсортированный вектор создается и возвращается в качестве вывода функции.

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

Чтобы загрузить описания, вы должны создать новый экземпляр класса Indexer и использовать имя файла в качестве аргумента в конструкторе

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

Будущие улучшения

Как новая библиотека, это много улучшений, и вот подробный список основных:

• Оптимизация времени запроса
• Поддержка с несколькими языками
• Улучшения в модели подписи
• Включить использование пользовательских моделей встраивания слов при поиске

Хотите помочь? Отправить запрос на слияние или электронное письмо на [email protected]