ZhengPeng7/BiRefNet

Что это?

Ключевые возможности

• 01Разделяет объекты переднего плана и фон на изображениях высокого разрешения.
• 02Предоставляет предварительно обученные веса для задач общего назначения, matting, портретного matting, DIS, HRSOD и COD.
• 03Позволяет загрузить модель одной строкой через Hugging Face Transformers.
• 04Включает конвертацию в ONNX и инференс в FP16 с опубликованными показателями скорости и потребления памяти.
• 05Поддерживает донастройку на пользовательских датасетах с помощью скриптов обучения и оценки.

Сценарии использования

• Удаление фона на фотографиях товаров и портретах высокого разрешения.
• Создание сервиса для удаления фона на изображениях или видео на основе модели PyTorch или ONNX.
• Донастройка модели сегментации на датасете для конкретной предметной области.
• Добавление удаления фона в рабочие процессы ComfyUI или Stable Diffusion в виде ноды.

Быстрый старт

1. 1

   Загрузите модель BiRefNet с Hugging Face с помощью transformers

   birefnet = AutoModelForImageSegmentation.from_pretrained('zhengpeng7/BiRefNet', trust_remote_code=True)

Что важно знать

• !Для практической работы с изображениями высокого разрешения обычно требуется GPU, поскольку показатели скорости в README измерялись на RTX 4090.
• !Согласно README проекта, для дальнейшего развития модели требуется больше ресурсов GPU, поэтому некоторые продвинутые варианты зависят от внешних вычислительных ресурсов.

Темы

Частые вопросы

Что такое BiRefNet?

Официальная реализация BiRefNet на PyTorch — модели для высокоточного разделения объектов и фона на изображениях, часто используемой для удаления фона. BiRefNet — модель глубокого обучения для разделения объектов и фона на изображениях высокого разрешения. Проект представляет собой официальную реализацию статьи, опубликованной в 2024 году в CAAI AIR, и включает скрипты на Python с использованием PyTorch для обучения, тестирования и оценки.

BiRefNet можно использовать бесплатно? Какая у него лицензия?

BiRefNet распространяется с открытым исходным кодом под лицензией MIT. Перед использованием в коммерческих продуктах проверьте условия лицензии.

BiRefNet всё ещё поддерживается?

Активно поддерживается. Последний коммит в BiRefNet — 17 окт. 2026 г..

Для чего можно использовать BiRefNet?

Удаление фона на фотографиях товаров и портретах высокого разрешения. Создание сервиса для удаления фона на изображениях или видео на основе модели PyTorch или ONNX. Донастройка модели сегментации на датасете для конкретной предметной области. Добавление удаления фона в рабочие процессы ComfyUI или Stable Diffusion в виде ноды.

Как установить BiRefNet?

Загрузите модель BiRefNet с Hugging Face с помощью transformers: birefnet = AutoModelForImageSegmentation.from_pretrained('zhengpeng7/BiRefNet', trust_remote_code=True)

Где скачать BiRefNet?

Исходный код и инструкции по установке — на GitHub: https://github.com/ZhengPeng7/BiRefNet