Как я обучил нейросеть генерировать постеры с Нивой

Всем привет! В жизни каждого мужчины наступает момент когда он хочет купить Ниву. Вот в 33 года и мне захотелось приобрести данный автомобиль. А пока я копил деньги на долгожданную покупку, захотелось какого нибудь мерча с Нивой. Быстрый поиск по популярным маркетплейсам не дал результатов, которые бы порадовали моего внутренного эстета.

Было решено сгенерировать изображения с помощью нейросети. Опробовал Dall‑E и Midjorney, Flux — все не то, нейросети никогда не видели нашей отечественной Нивы, получались жалкие пародии, больше похожие на короткий Land Rover или Toyota. Пример того что выдавали нейросети разместил ниже.

Подумав что я же инженер, решил дообучить Flux на открытых фотографиях Нивы из интернета.

Обучение

Обучение оказалось до неприличного простым. Я использовал сервис replicate.com для запуска и обучения Flux. У них есть прекрасный туториал на английском как дообучать Flux. Опишу тут кратко на русском для тех кому лень читать.

1. Подбираем картинки для обучения. Примерно 10–20 изображений в высоком качестве

2. (Опционально) Для каждого изображения пишем текстовое описание и сохраняем его в файл с тем же именем, но расширением (.txt)

3. Складываем все в zip архив и загружаем в сервис replicate

4. Ждем 20–30 минут пока нейросеть дообучится

5. Генерируем изображения использую триггер слово (в моем случае NIVA)

Результаты

Я был удивлен что примерно за 20 минут нейросеть дообучилась и начала выдавать красивые изображения Нивы. Помню раньше требовалось намного больше усилий — развертывание модели у себя на компьютере, борьба с разными версиями библиотек…

Для генерации изображений я использовал следующий промпт:

Imagine a sleek and modern t-shirt design featuring a powerful NIVA as the central element. NIVA should be depicted in a dynamic and adventurous pose, capturing the essence of freedom and exploration. Consider incorporating elements like rugged terrain or a scenic background to enhance the outdoor theme. The color palette should be bold and vibrant, reflecting the spirit of adventure. Aim for a design that is both stylish and eye-catching, appealing to NIVA enthusiasts and adventure.

В процессе поиска идеального постера для футболки сгенерировал примерно 20–30 изображений. Стоимость обучения и генерации составила примерно $5.

Спасибо что дочитали до конца! Буду рад вашим комментариям.