Нейросеть в память о Ките Харинге
Описание проекта
У меня было несколько целей, чтобы обучить нейросеть рисовать в определенной стилистике:
Первая. Меня нужно научиться это делать, чтобы потом сделать иллюстрации для книжки в стиле иллюстратора, который пропал.
Вторая чуть более интересная. Мне очень нравится художник Кит Харинг, но он умер ещё до моего рождения. Когда я был в Берлине, я узнал, что на берлинской стене была одна из самых больших его работ, но и её разрушили до моего рождения, поэтому и этого я никогда не увижу. Поэтому я хочу увидеть, как бы выглядели цветные фотографии берлинской стены с рисунками этого удивительного художника.
Берлинская стена со знаменитыми «человечками»
Так я не только научусь использовать новые технологии, но и выполню несколько задач.
Исходные изображения для обучения
Результат обучения нейросети
Анализ изображений
Методом проб и ошибок я понял, что нейросеть плохо справляется с пониманием точной формы элементов. Но это можно легко исправить при помощи Photoshop, Illustrator или Procreate. Например, вырезая объект из фона можно добиться желаемого результата.
" an art in KEITHHA style, yellow dog with red heart in his mouth, blue background "
Код и описание процесса обучения
ссылка на код https://www.kaggle.com/code/vasyachukovskiy/keithha
1. Аутентификация и подготовка окружения
В начале происходит вход в аккаунт Hugging Face с помощью notebook_login (), что позволяет загружать и сохранять модели в репозитории
2. Запуск обучения
С помощью команд проверки и загрузки базы изображений запускаем код.
Для ускорения процесса обучения выставлено всего 500 шагов с двумя чекпоинтами (250 и 500).
3. Процесс обучения и сохранения промежуточных результатов
Во время обучения модель проходит через фазы прямого и обратного прохода. Процесс не быстрый, но с ускорением занимает не так много времени.
4. После завершения обучения производится сбор итоговой модели
Сохраняем обученную нейросеть на HuggingFace.co
Кит Харинг с другой стороны
