Bria AI выпустили RMBG 2.0 — нейросеть для удаления фона на фотографиях

Разработчики организации Bria AI выпустили RMBG 2.0 — обновление нейросети для удаления фона на фотографиях. Модель машинного обучения теперь лучше распознаёт объекты на переднем плане и поддерживает больше сценариев работы.

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

За основу авторы проекта взяли архитектуру нейросети BiRefNet, но улучшили метод обучения и использовали собственный датасет. Это позволило добиться результата лучше, чем у оригинальной модели.

Нейросеть RMBG опубликовали на площадке Hugging Face. Есть версии для PyTorch и Safetensors. В тестовом пространстве разработчики развернули бесплатное демо. Для некоммерческого использования RMBG доступна по лицензии Creative Commons. 

Пример кода для запуска: