Обучение нейросети
Идея проекта
Идея проекта заключается в улучшении качества старых фотографий и возможность генерировать другие виды, для облегченной генерации идей. Для основы своей идеи я решила взять «ежика в тумане» и начала генерировать новые изображения благодаря нейросети
Проверка, что подключенный видеоускоритель работает и подключен
Следующие ячейки отвечают за установку нужных библиотек и код тренировки нейросети
далее из датасета переносим в созданную папку в корне проекта и делаем функцию для отображения изображений
отображаем все изображения из нашей созданной папки
создание экземпляра нейросети для определения картинок и создание подсказки для наших изображений
тут нейросеть сканирует изображения и создает подсказки чтобы нейросеть понимала что изучать на изображении
само обучение нейросети, настройки шагов, промпта и т. д.
следующие две ячейки отвечают за загрузку обученной нейросети в hugging face
далее подключаемся к ней и генерируем уже наши изображения
Файл с кодом