BiRefNet 安装和配置指南

BiRefNet 安装和配置指南

BiRefNet [arXiv'24] Bilateral Reference for High-Resolution Dichotomous Image Segmentation 项目地址: https://gitcode.com/gh_mirrors/bi/BiRefNet

1. 项目基础介绍和主要编程语言

项目基础介绍

BiRefNet 是一个用于高分辨率二值图像分割的开源项目，由 ZhengPeng7 开发。该项目在 CAAI AIR 2024 会议上正式发布，旨在通过双边参考技术提升图像分割的性能。BiRefNet 在多个高分辨率任务上达到了最先进的水平，包括 DIS、COD 和 HRSOD 等。

主要编程语言

该项目主要使用 Python 编程语言进行开发。

2. 项目使用的关键技术和框架

关键技术和框架

• PyTorch: 作为深度学习框架，用于模型的训练和推理。
• ONNX: 用于模型的转换和优化，提升推理效率。
• Hugging Face: 提供了模型的预训练权重和在线推理接口。
• TensorRT: 用于进一步优化模型的推理速度。

3. 项目安装和配置的准备工作和详细安装步骤

准备工作

在开始安装之前，请确保您的系统满足以下要求：

• Python 3.7 或更高版本
• CUDA 11.8 或更高版本（如果使用 GPU）
• PyTorch 2.0.1 或更高版本
• Git

详细安装步骤

步骤 1: 克隆项目仓库

首先，使用 Git 克隆 BiRefNet 项目到本地：

git clone https://github.com/ZhengPeng7/BiRefNet.git
cd BiRefNet

步骤 2: 创建虚拟环境（可选）

为了隔离项目依赖，建议创建一个虚拟环境：

python -m venv birefnet_env
source birefnet_env/bin/activate  # 在 Windows 上使用 birefnet_env\Scripts\activate

步骤 3: 安装依赖

安装项目所需的 Python 依赖包：

pip install -r requirements.txt

步骤 4: 下载预训练模型

您可以从 Hugging Face 下载预训练的模型权重：

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

步骤 5: 运行示例代码

项目中提供了一些示例代码，您可以运行这些代码来测试安装是否成功：

python train.py  # 训练模型
python inference.py  # 进行推理

步骤 6: 配置 ONNX 和 TensorRT（可选）

如果您希望进一步提升推理速度，可以配置 ONNX 和 TensorRT：

1. ONNX 转换: 参考项目中的 tutorials/onnx_conversion.ipynb 进行 ONNX 转换。
2. TensorRT 配置: 参考 yuanyang1991/birefnet_tensorrt 项目进行 TensorRT 配置。

总结

通过以上步骤，您应该能够成功安装和配置 BiRefNet 项目。如果在安装过程中遇到任何问题，请参考项目的 GitHub 页面或联系项目维护者获取帮助。

BiRefNet [arXiv'24] Bilateral Reference for High-Resolution Dichotomous Image Segmentation 项目地址: https://gitcode.com/gh_mirrors/bi/BiRefNet