Dezign

Концепция

Я с детства искренне не понимала, почему именно черные коты стали символом неудачи, спутниками всякого рода зла и нечести. Почему образ, который мог бы быть просто красивым и загадочным, обернулся для многих символом тревоги? Мне всегда было искренне жаль ни в чем неповинных котиков. Эта детская обида и стала отправной точкой проекта: мне захотелось дать им новую, собственную визуальную историю, в которой не будет места страху, но будет место красоте, характеру и, главное, домашнему теплу.

Много котиков, как известно, не бывает, поэтому я решила, что могу нагенерировать ещё парочку черных красавцев — таких, какими вижу их сама.

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

Начало проекта. Датасет

Исходный размер 2048x512

В качестве датасета я взяла фотографии разных котиков в разной обстановке. Обрезала их до квадрата.

Самое сложное! Код!

Исходный размер 886x481

Для обучения модели я использовала специальный скрипт DreamBooth. В нём я указала путь к заранее подготовленным фотографиям котиков, задала настройки обучения — например, что картинки будут обрабатываться в разрешении 512 пикселей, а сам процесс займёт 250 шагов. Все эти параметры помогли модели быстро и аккуратно «запомнить», как выглядят именно мои чёрные коты, не перегружая память компьютера (на 500 шагах ноутбук сдавался на половине процесса и вел себя неприлично).

Исходный размер 1059x299

Здесь я подготавливаю нейросеть к созданию изображений. Сначала я загружаю специальную улучшенную версию кодировщика VAE (он отвечает за чёткость деталей и стабильность цветов). Его я взяла от разработчика madebyollin, потому что эта версия работает быстрее и экономичнее памяти. Затем я собираю основную модель Stable Diffusion XL в единый pipeline — это главный инструмент, который будет генерировать картинки. Я указываю, что работаю с 16-битной точностью, чтобы ускорить процесс и снизить нагрузку на видеокарту, а также использую безопасный формат safetensors для хранения весов модели — это надёжнее, чем обычные файлы. Всё вместе это позволяет быстро и стабильно запускать генерацию прямо в Colab.

Исходный размер 883x742

В конце я загружаю в модель результаты обучения (LoRA), которая запомнила образ кота. После загрузки я переношу модель на видеокарту (CUDA), чтобы генерация шла быстро. Затем я задаю промты, по которым нейросеть будет генерировать. Для каждого промта модель делает 30 шагов уточнения, а параметр guidance_scale отвечает за то, насколько строго она будет следовать моему описанию.

Результаты генерации

Исходный размер 1280x256

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

Однако, пара из них, как мне кажется, прекрасно подходит под создание мемов, так что я решила сделать два, вдохновленные истинным зумерским юмором.