使用方法 (Transformers.js)

如果您还没有安装，可以通过 NPM 安装 Transformers.js JavaScript 库：

npm i @huggingface/transformers

示例： 使用 onnx-community/mediapipe_selfie_segmentation-web 进行自拍分割。

import { AutoModel, RawImage, Tensor } from '@huggingface/transformers';

// 加载模型和处理器
const model_id = 'onnx-community/mediapipe_selfie_segmentation-web';
const model = await AutoModel.from_pretrained(model_id, { dtype: 'fp32' });

// 从 URL 加载图像
const url = 'https://huggingface.co/datasets/Xenova/transformers.js-docs/resolve/main/selfie_segmentation.png';
const image = await RawImage.read(url);

// 预测 alpha 蒙版
const { alphas } = await model({
    pixel_values: new Tensor(
        'uint8',
        image.data,
        [1, image.height, image.width, 3],
    ),
});

// 保存输出蒙版
const mask = RawImage.fromTensor(alphas[0].mul(255).to('uint8'), 'HWC')
mask.save('mask.png');

// (可选) 将蒙版应用到原始图像
const result = image.clone().putAlpha(mask);
result.save('result.png');

输入图像	预测蒙版	输出图像