Xenova/segformer-b0-finetuned-ade-512-512

来自 https://huggingface.co/nvidia/segformer-b0-finetuned-ade-512-512 的 ONNX 权重，与 Transformers.js 兼容。

使用方法 (Transformers.js)

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

npm i @huggingface/transformers

示例： 使用 Xenova/segformer-b0-finetuned-ade-512-512 进行图像分割。

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

// 创建图像分割管道
const segmenter = await pipeline('image-segmentation', 'Xenova/segformer-b0-finetuned-ade-512-512');

// 对图像进行分割
const url = 'https://huggingface.co/datasets/Xenova/transformers.js-docs/resolve/main/house.jpg';
const output = await segmenter(url);
console.log(output)
// [
//   {
//     score: null,
//     label: 'wall',
//     mask: RawImage { ... }
//   },
//   {
//     score: null,
//     label: 'building',
//     mask: RawImage { ... }
//   },
//   ...
// ]

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

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

注意：为 ONNX 权重单独创建一个仓库是一个临时解决方案，直到 WebML 获得更多关注。如果你希望让你的模型可以在 Web 上使用，我们推荐使用 🤗 Optimum 转换为 ONNX 格式，并像此仓库一样组织（ONNX 权重位于名为 onnx 的子文件夹中）。