В мире генерации изображений с помощью ИИ, таких как Stable Diffusion, точные промпты играют ключевую роль. Image Prompt — это простое десктопное приложение, которое анализирует загруженные изображения и генерирует списки тегов, описывающих их содержимое. Эти теги можно использовать как основу для промптов, чтобы создавать похожие изображения или улучшать поиск в коллекциях. Приложение работает локально, поддерживает несколько моделей ИИ и предлагает гибкие настройки для фильтрации результатов. Ниже разберём функционал шаг за шагом.

Основные возможности

Загрузка и просмотр изображений

Приложение открывается с интерфейсом в тёмной теме: слева — область для изображения, справа — настройки и вывод. Поддерживаются форматы PNG, JPG, JPEG, BMP, GIF. Изображение загружается перетаскиванием в специальный блок или через кнопку "Browse Image". Возможна загрузка по URL из интернета. Если изображение превышает 1024x1024 пикселей, оно автоматически масштабируется для обработки, но оригинал остаётся неизменным. В блоке отображается предпросмотр.

Выбор модели тегирования

В разделе настроек доступен список моделей на базе нейронных сетей:

  • WD14: Универсальная модель для общего анализа (сцены, объекты, цвета).
  • DeepDanbooru: Ориентирована на аниме-стиль (теги вроде "anime girl, long hair").
  • E621: Специализирована на фурри и аниме-контенте.
  • WD SwinV2 v3, WD VIT v3, WD Convnext v3, WD EVA02 v3: Улучшенные версии WD для более точного распознавания деталей и персонажей.

Модели загружаются автоматически с Hugging Face при первом использовании. Каждая генерирует от десятка до сотен тегов, отсортированных по релевантности. Смена модели запускает перегенерацию тегов.

Настройки фильтрации и точности

Раздел "Settings" позволяет кастомизировать вывод:

  • Пороги уверенности: Слайдеры "General Threshold" (по умолчанию 0.35) и "Character Threshold" (по умолчанию 0.90) отсеивают теги с низкой вероятностью. Первый — для общих описаний, второй — для персонажей.
  • Включение персонажей: Чекбокс "Use Characters" добавляет теги вроде "1girl" или имён персонажей.
  • Запрещённые теги: Текстовое поле для ввода слов/фраз (через запятую или строки), которые исключаются из результата (например, "blurry, low quality"). Проверка нечувствительна к регистру.
  • Замена тегов: Чекбокс "Use Tag Replacement" активирует поля для замены. Введите теги для замены (например, "1girl, blue eyes") и их аналоги (например, "woman, mature; green eyes"). Это позволяет адаптировать теги под конкретные нужды.

Настройки применяются на лету, фильтруя сырые теги от модели.

Стили тегов и промптов

  • Tag Style: Выбор из "Custom" (чистый список), "Illustrious" (добавляет префикс "masterpiece, best quality...") или "Pony" (префикс "score_9, score_8_up...") для оптимизации под популярные ИИ-модели.
  • Prompt Style: Список стилей из файла "styles_integrated.txt" (можно редактировать). Добавляет готовые фразы вроде "highly detailed, cinematic lighting", заменяя {prompt} на теги. Кнопка "Refresh Presets" обновляет список.

Результат — готовый промпт в блоке "Generated Tags", комбинирующий теги и стиль.

Генерация и вывод результатов

Кнопка "Generate Tags" запускает анализ (или это происходит автоматически при изменениях). Процесс отображается в прогресс-баре и логах ("Logs"). На выходе — текст с тегами, который можно скопировать кнопкой "Copy Tags". Кнопка "Clear" очищает всё.

Дополнительно

  • Настройки сохраняются в config.json для удобства.
  • Нет батч-обработки — только одно изображение за раз.
  • Интерфейс: Раздвижные панели, тёмная тема для комфорта.

Image Prompt идеален для художников, ИИ-энтузиастов и коллекционеров изображений.