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

2025-09-10

1.8 K

BEN2本地部署全流程指南

環境準備：需Python 3.8+环境，建议配备NVIDIA显卡（支持CUDA加速）

モデルの入手::

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

モデルの初期化::

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

経営推論::

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

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

関連文書ダウンロードアドレス

このリソースをダウンロードするにはログインが必要です。サインイン

このサイト上のすべてのリソースは、学習目的のためにのみ、ネットワークからのものです！