Image Segmentation
Transformers
ONNX
Transformers.js
English
u2net
isnet
dis
mask-generation
vision
background-removal
portrait-matting
Instructions to use BritishWerewolf/IS-Net with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use BritishWerewolf/IS-Net with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-segmentation", model="BritishWerewolf/IS-Net")# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("BritishWerewolf/IS-Net", device_map="auto") - Transformers.js
How to use BritishWerewolf/IS-Net with Transformers.js:
// npm i @huggingface/transformers import { pipeline } from '@huggingface/transformers'; // Allocate pipeline const pipe = await pipeline('image-segmentation', 'BritishWerewolf/IS-Net'); - Notebooks
- Google Colab
- Kaggle
File size: 520 Bytes
3e77535 9783722 3e77535 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 | {
"_name_or_path": "BritishWerewolf/IS-Net",
"model_type": "u2net",
"architectures": [
"U2NetModel"
],
"transformers.js_config": {
"dtype": "fp32"
},
"input_name": "input_image",
"input_shape": [1, 3, 1024, 1024],
"output_composite": "output_image",
"output_names": [
"1890",
"1891",
"1892",
"1893",
"1894",
"input.656",
"input.740",
"input.832",
"input.948",
"input.1088",
"input.1252",
"output_image"
],
"output_shape": [1, 1024, 1024]
}
|