https://huggingface.co/facebook/maskformer-swin-tiny-ade 的 ONNX 权重，用于兼容 Transformers.js。

使用方法 (Transformers.js)

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

npm i @huggingface/transformers

示例： 使用 onnx-community/maskformer-swin-tiny-ade 进行场景分割。

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

// 创建图像分割流水线
const segmenter = await pipeline('image-segmentation', 'onnx-community/maskformer-swin-tiny-ade');

// 对图像进行分割
const url = 'https://huggingface.co/datasets/hf-internal-testing/fixtures_ade20k/resolve/main/ADE_val_00000001.jpg';
const output = await segmenter(url);
console.log(output)
// [
//   {
//     score: 0.9240802526473999,
//     label: 'plant',
//     mask: RawImage { ... }
//   },
//   {
//     score: 0.967036783695221,
//     label: 'house',
//     mask: RawImage { ... }
//   },
//   ...
//   }
// ]

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

for (let i = 0; i < output.length; ++i) {
  const { mask, label } = output[i];
  mask.save(`${label}-${i}.png`);
}

注意：为 ONNX 权重创建单独的仓库是临时解决方案，直到 WebML 获得更广泛的支持。如果您想让您的模型支持 Web 端使用，我们建议使用 🤗 Optimum 转换为 ONNX 格式，并按照此仓库的结构组织（将 ONNX 权重放在名为 onnx 的子文件夹中）。