py search images

CLI Tool: Поиск изображения с использованием зажима и векторных вторжений

Этот инструмент CLI позволяет индексировать изображения из указанной папки и искать аналогичные изображения на основе текстового запроса с использованием CLIP (контрастное представление о языке), Pytorch и Numpy. Он генерирует векторы изображения и позволяет эффективно выполнять семантический поиск.

Функции

• Изображения индекса и генерировать векторные вторжения.
• Поиск изображений на основе текстовых запросов.
• Поддерживает несколько форматов изображений (.jpg, .jpeg, .png, .webp).

Использование

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

Индексные изображения

Чтобы индексировать изображения из images папок, запустите:

(venv) ➜ vector-embeddings python main.py index -f images

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

Поиск соответствующих изображений

Чтобы найти аналогичные изображения на основе текстового запроса, запустите:

(venv) ➜ vector-embeddings python main.py search " Your text query " -f images

Замените «ваш текстовый запрос» на описание изображения, которое вы ищете. Инструмент загрузит векторы из ectors.npz и вернет наиболее похожие изображения.