说明文档

用法 (Transformers.js)

如果你还没有安装，可以从 NPM 安装 Transformers.js JavaScript 库：

npm i @huggingface/transformers

示例： 使用 onnx-community/sapiens-seg-0.3b 进行语义分割。

import { pipeline } from '@huggingface/transformers';

const segmenter = await pipeline('image-segmentation', 'onnx-community/sapiens-seg-0.3b');

const url = 'https://huggingface.co/datasets/Xenova/transformers.js-docs/resolve/main/ryan-gosling.jpg';
const output = await segmenter(url);
console.log(output)
// [
//   {
//     score: null,
//     label: 'Background',
//     mask: RawImage { ... }
//   },
//   {
//     score: null,
//     label: 'Apparel',
//     mask: RawImage { ... }
//   },
//   ...
// ]

你可以使用以下代码可视化输出结果：

for (const l of output) {
  l.mask.save(`${l.label}.png`);
}

onnx-community/sapiens-seg-0.6b

作者 onnx-community

image-segmentation transformers.js

↓ 1 ♥ 1

创建时间: 2024-08-26 00:09:56+00:00

更新时间: 2025-03-06 17:19:33+00:00

在 Hugging Face 上查看

文件 (13)

.gitattributes

README.md

config.json

onnx/model.onnx ONNX

onnx/model.onnx_data

onnx/model_bnb4.onnx ONNX

onnx/model_fp16.onnx ONNX

onnx/model_int8.onnx ONNX

onnx/model_q4.onnx ONNX

onnx/model_q4f16.onnx ONNX

onnx/model_quantized.onnx ONNX

onnx/model_uint8.onnx ONNX

preprocessor_config.json