Обучение генеративной нейросети для создания Вышкинской вороны
Идеей проекта является обучение нейросети Stable Diffusion генерировать изображения Вышкинской вороны
Dataset
Dataset
Для начала подключаю собранные картинки к гугл диску
Далее устанавливаю, что все картинки должны быть в итоге размером 1024×1024 пикселей и сохраняю новые изображения в папку в формате JPG с хорошим качеством
Эта часть кода создает сетку из изображений, расположенных в указанной папке, и позволяет предварительно просмотреть первые несколько изображений
А вот и они
Далее указываю папку, с которой буду работать
Сгенерировала префиксы, чтобы у каждого изображения появился промпт
После этого я зарегистрировалась на Hugging Face и получила токен
С первого раза мне не удалось обучить модель, она выдала ошибку, поэтому я обратилась к Chat GPT. Он посоветовал увеличить время задержки скачивания программу для обучения
Результат
