Leffa: открытый инструмент Meta AI для управляемой генерации изображений людей, идеально подходит для виртуальной примерки 190

小编 ·

Leffa: открытый инструмент Meta AI для управляемой генерации изображений людей, идеально подходит для виртуальной примерки

Leffa (Learning Flow Fields in Attention) — технология генерации изображений, разработанная и открытая Meta AI. Leffa представляет собой единый фреймворк, который с помощью обучения потоковым полям (flow fields) в attention направляет целевые запросы к нужным областям референсного изображения во время обучения. Вводя функцию регуляризационных потерь в базовую модель на основе диффузии, метод значительно снижает потерю деталей, сохраняя высокое качество генерации. Технология подходит не только для виртуальной примерки, но и...

Leffa: открытый инструмент Meta AI для управляемой генерации изображений людей, идеально подходит для виртуальной примерки

Что такое Leffa?

Leffa, полное название Learning Flow Fields in Attention, — это технология генерации изображений, разработанная и открытая компанией Meta ai. Leffa представляет собой унифицированный фреймворк, который, обучая поля потоков (flow fields) во внимании, направляет целевые запросы на корректное внимание к ключевым областям эталонного изображения во время обучения. Этот метод, вводя функцию потерь регуляризации в базовую модель на основе диффузии, значительно уменьшает искажение деталей, сохраняя при этом высокое качество генерации изображений. Эта технология применима не только для виртуальной примерки, но и поддерживает перенос поз, а также может быть расширена на другие модели диффузии.

Leffa.webp

Ключевые особенности Leffa

  • Точный контроль: Leffa позволяет осуществлять виртуальную примерку (естественное применение изображений конкретной одежды к человеку) и перенос поз (перенос позы одного человека на другое изображение), сохраняя при этом исходные черты и чёткость деталей человека.

  • Высокое качество генерации: Благодаря функции потерь регуляризации и стратегии постепенного обучения, Leffa оптимизирует производительность модели, обеспечивая высокое разрешение и детализацию генерируемых изображений, уменьшая размытие текстур и проблемы деформации. Например, при генерации изображений человека в определённой одежде Leffa отлично сохраняет текстуру и узоры одежды.

  • Независимость от модели и обобщающая способность: Leffa спроектирован так, чтобы интегрироваться в различные модели диффузии без дополнительных параметров, что свидетельствует о хорошей универсальности и адаптивности, позволяя применять его во множестве сценариев генерации изображений ИИ.

  • Высокая эффективность: На высокопроизводительном оборудовании, таком как A100 GPU, Leffa способен генерировать высококачественные изображения человека за 6 секунд, поддерживает ускорение float16 и подходит для задач, требующих быстрой генерации.

  • Лёгкость развёртывания: Поддерживает локальное развёртывание, облачный инференс и интеграцию с такими платформами, как Hugging Face, что позволяет разработчикам и компаниям быстро интегрировать его в существующие рабочие процессы ИИ.

Leffa框架的技术原理.webp

Технические принципы Leffa

  • Механизм внимания: Основан на механизме внимания, связывающем целевое изображение (изображение человека для генерации) с эталонным изображением (предоставляющим внешний вид или позу) посредством слоёв внимания.

  • Обучение полей потоков: Основано на обучении полей потоков (flow fields) в слоях внимания, явно направляя целевой запрос (target query) на правильные области эталонного ключа (reference key).

  • Потери регуляризации: Применяет потери регуляризации к картам внимания, деформируя эталонное изображение для более точного совмещения с целевым, побуждая модель во время обучения правильно фокусироваться на эталонных областях.

  • Пространственная согласованность: Основано на преобразовании карт внимания в поля потоков, деформируя эталонное изображение с помощью операции сэмплирования по сетке, обеспечивая пространственную согласованность между целевым запросом и эталонным изображением.

  • Независимость от модели: В роли функции потерь регуляризации интегрируется в различные модели диффузии без дополнительных параметров или сложных техник обучения.

  • Постепенное обучение: Применяется на завершающем этапе обучения, избегая ранней деградации производительности, основано на комбинировании традиционных диффузионных потерь и потерь Leffa для тонкой настройки, оптимизируя производительность модели.

Leffa的应用场景.webp

Сценарии применения Leffa

  • Виртуальная примерка: В электронной коммерции и модной индустрии — создание виртуальных примерочных, позволяющих потребителям увидеть, как на них будет сидеть различная одежда, без необходимости физической примерки.

  • Дополненная реальность (AR): В приложениях AR — изменение или добавление внешности и одежды пользователя в реальном времени, обеспечивая более иммерсивный опыт.

  • Игры и развлечения: В разработке игр — для настройки персонажей, позволяя игрокам настраивать внешность и позы персонажей по своему усмотрению.

  • Производство фильмов и видео: В постпродакшене фильмов — для генерации или модификации образов персонажей.

Информация об открытом исходном коде Leffa

  • Статья: https://arxiv.org/pdf/2412.08486

  • Репозиторий на GitHub: https://github.com/franciszzj/Leffa

  • Онлайн-демо: https://huggingface.co/spaces/franciszzj/Leffa

  • Библиотека моделей на HuggingFace: https://huggingface.co/franciszzj/Leffa