GHOST 2.0: Новый уровень крутости в AI и фотомоделировании!

🔥 У нас просто зашкаливает уровень крутости! 🥳 Наши талантливые коллеги запустили open-source модель для переноса головы на фотографии — встречайте GHOST 2.0! 📸 Данный искусственный интеллект можно найти на всех лучших вечеринках tech-мира, и он берет на себя важные задачи!

Мы хоть и были на заднем плане, но все равно внесли свой вклад, помогая друзьям с обучением моделей в области портретной сегментации лиц и фейс парсинга. Это как защита диплома, только мы его сделали без лишних волнений! 😎

Что же нужно для работы с GHOST? Наш герой требует, чтобы модель сегментации отвечала нескольким важным аспектам. Вот они:

Сегментация причесок и волос на лице как отдельные классы — представьте, что ваши волосы — это независимые личные помощники, а прическа — ваш личный стилист! 💁‍♂️💁‍♀️
Аддитивность модели — такая штука, что мы можем комбинировать сегменты и составлять полные головы, как кузнецы своего дела! 🧙‍♂️
Однородность цветов каждого региона — класс «борода» не должен путаться с классом «кожа». Тут все по честному, без пересечений! 😂

Работа была проведена с использованием датасета EasyPortrait, который состоит из 20,000 HD+ изображений реальных людей. Представляете, сколько историй за каждым из них? 🤔 Мы выбрали для обучения модель Segformer-B5 — она явно знала, что делать, и не подвела!

На графике ниже вы можете увидеть визуальные сравнения нашей модели с базовой — это как участие в соревнованиях, где мы показали класс! 🏆

И чтобы не оставлять вас без полезных ссылок в конце ленты, вот они: 👉 ArXiv – для тех, кто хочет заглянуть под капот! 👉 Habr – техно-новости, как они есть! 👉 Repo – для всех, кто готов играть с GHOST дальше!

Кажется, мы движемся к великому будущему, и GHOST 2.0 — это только начало! 🚀✨