如何在自己的电脑上部署BEN2进行本地图像处理？

2025-09-10

1.8 K

BEN2本地部署全流程指南

Preparação ambiental：需Python 3.8+环境，建议配备NVIDIA显卡（支持CUDA加速）

Obtendo o modelo::

git clone https://huggingface.co/PramaLLC/BEN2
cd BEN2

Instalação de dependências: Implementaçãopip install -r requirements.txt（包含PyTorch等核心库）

Inicialização do modelo::

from model import BEN_Base
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
model = BEN_Base().to(device).eval()

raciocínio executivo::

mask, foreground = model.inference(Image.open('test.jpg'))
mask.save('result_mask.png')

advertência：首次运行会自动下载1.13GB模型文件；4K处理建议至少有8GB显存；CPU模式处理时间约为GPU的3-5倍。