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
| { | |
| "_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] | |
| } | |