Views
No views yet
npm i @huggingface/transformersXenova/segformer-b0-finetuned-cityscapes-1024-1024.1import { pipeline } from '@huggingface/transformers';
2
3// Create an image segmentation pipeline
4const segmenter = await pipeline('image-segmentation', 'Xenova/segformer-b0-finetuned-cityscapes-1024-1024');
5
6// Segment an image
7const url = 'https://huggingface.co/datasets/Xenova/transformers.js-docs/resolve/main/cityscapes.png';
8const output = await segmenter(url);
9console.log(output);
10// [
11// {
12// score: null,
13// label: 'road',
14// mask: RawImage { ... }
15// },
16// {
17// score: null,
18// label: 'sidewalk',
19// mask: RawImage { ... }
20// },
21// ...
22// ]1for (const l of output) {
2 l.mask.save(`${l.label}.png`);
3}onnx).