neuro-sketch

0

Описание

Простое приложение для превращения ручных скетчей в реалистичные фотографии с помощью AI

Языки

  • Python100%
8 месяцев назад
8 месяцев назад
8 месяцев назад
8 месяцев назад
8 месяцев назад
README.md

🎨 NeuroSketch - AI Sketch to Photo Converter

Превращайте ваши ручные рисунки в реалистичные фотографии с помощью искусственного интеллекта!

Python PyTorch

✨ Возможности

  • 🖌️ Интуитивный интерфейс - рисуйте мышкой прямо в приложении
  • 🤖 AI генерация - использует ControlNet + Stable Diffusion v1.5
  • 🎯 Настраиваемый промпт - опишите желаемый результат
  • 🔄 Гибкость моделей - легко заменяйте модели на другие
  • Быстрая генерация - в зависимости от вашей видеокарты 1-5 минут
  • 🖼️ Размер изображения - 512x512 пикселей

🖼️ Примеры работ

Пример 1: Животное

CкетчРезультат
Скетч животногоРезультат животного

Примечание: Изображения выше - примеры. Ваши результаты могут отличаться в зависимости от качества скетча и промпта. Надеюсь вы рисуете лучше меня

🚀 Установка

Требования

  • Python 3.10+
  • CUDA-совместимая видеокарта (рекомендуется)
  • 4+ ГБ свободного места на диске

Шаги установки

  1. Клонируйте репозиторий
  1. Установите зависимости

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

  1. Запустите приложение

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

  1. Нарисуйте скетч

    • Используйте мышь для рисования на белом холсте
  2. Введите промпт

    • Опишите желаемый результат в поле для промпта или оставьте его по-умолчанию
  3. Сгенерируйте изображение

    • Нажмите кнопку "🎨 Сгенерировать"
    • Дождитесь завершения (скорость зависит от видеокарты, на процессоре не рекомендую)
  4. Получите результат

    • Изображение автоматически откроется
    • Сохранится как
      output.png

🛠️ Технические детали

Используемые технологии

  • Stable Diffusion v1.5 - базовая модель генерации
  • ControlNet Canny - обработка контуров скетча
  • Tkinter - графический интерфейс

Архитектура

Скетч → Canny Edge Detection → ControlNet → Stable Diffusion → Результат

📁 Структура проекта

neuro-sketch/ ├── inference.py # Основное приложение ├── examples/ # Папка с примерами работ │ ├── drawing.png # Скетч │ └── output.png # Результат └── README.md # Этот файл

📄 Лицензия

MIT License - используйте свободно для любых целей!

🤝 Вклад в проект

Приветствуются:

  • 🐛 Исправления багов
  • ✨ Новые функции
  • 📚 Улучшение документации
  • 🎨 Примеры использования

📞 Поддержка

Если у вас есть вопросы или проблемы:

  • Создайте Issue в GitHub
  • Опишите проблему подробно
  • Приложите скриншоты при необходимости

Создано с ❤️ для творческих людей! За один вечер, так что работает и на том спасибо