Derzeitige Position:Abb. Anfang " AI-Antworten

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

2025-09-10

1.8 K

BEN2本地部署全流程指南

Vorbereitung der Umwelt：需Python 3.8+环境，建议配备NVIDIA显卡（支持CUDA加速）

Das Modell erhalten::

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

Installation von Abhängigkeiten: Umsetzungpip install -r requirements.txt（包含PyTorch等核心库）

Initialisierung des Modells::

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

exekutive Argumentation::

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

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