29.03.2024
Подписывайтесь на Telegram-канал по ссылке

Image Inpainting от NVIDIA реалистично отредактирует фотографии с помощью ИИ

Image Inpainting от NVIDIA

Сервис Image Inpainting, разработанный компанией NVIDIA, ретуширует изображения при помощи «умной» кисти. Он работает на базе графических процессоров NVIDIA и использует алгоритмы глубокого обучения.

Новый инструмент NVIDIA восстанавливает повреждённые изображения с недостающими пикселями. Он также способен редактировать фотографии: удалять их часть и затем реалистично наполнять. Технология позволяет обрабатывать «пустоты» любой формы, размера и расстояния от границ изображения, считают разработчики.

При обучении нейронной сети специалисты использовали графические процессоры NVIDIA Tesla V100 и фреймворк машинного обучения PyTorch. Для этого разработчики сгенерировали 55 тысяч случайных «полос» и «пустот», которые наложили на изображения из датасетов ImageNet, Places2 и CelebA-HQ. Так они научили нейронную сеть восстанавливать недостающие пиксели. На этапе тестирования на те же изображения накладывали «полосы» и «пустоты», которые не использовались на стадии обучения. Это позволило проверить точность и аккуратность реконструкции.

По словам разработчиков, Image Inpainting можно использовать и для обработки фотографий с помощью технологии Super Resolution.