MathiasGruber/PConv-Keras
📦 Open Source ProjectMathiasGruber
Неофициальная реализация частичных сверток (Partial Convolutions) на Keras для качественного заполнения произвольных областей на изображениях.
Данный репозиторий предлагает решение глубокого обучения для инпейнтинга (inpainting), решающее проблему заполнения произвольных областей, с которыми стандартные сверточные слои справляются плохо. В отличие от обычных сверток, которые обрабатывают все пиксели одинаково, частичные свертки (PConv) используют маскированную и нормализованную операцию, где маска обновляется на каждом слое. Это позволяет нейросети различать валидные пиксели и «дыры», что приводит к созданию более четких и логичных результатов.
Реализация включает архитектуру для обновления масок, что делает её эффективной для реставрации фото, удаления объектов и аугментации данных. Проект построен на API Keras с бэкендом TensorFlow и обладает модульной структурой, позволяющей экспериментировать с глубиной сети и конфигурациями обучения. Проект широко известен своей доступностью, позволяя разработчикам применять передовые исследования NVIDIA в реальных задачах без необходимости реализации сложных математических компонентов с нуля.
💡Основное
- ├─Архитектура частичных сверток
- ├─Поддержка масок произвольной формы
- └─Совместимость с Keras и TensorFlow
🎯Для
- ├─Инженеры по компьютерному зрению
- ├─AI-исследователи
- └─Разработчики систем обработки изображений