Instructions to use netradrishti/u2net-saliency with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use netradrishti/u2net-saliency with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("feature-extraction", model="netradrishti/u2net-saliency", trust_remote_code=True)# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("netradrishti/u2net-saliency", trust_remote_code=True, device_map="auto") - Notebooks
- Google Colab
- Kaggle
| { | |
| "image_processor_type" : "U2NetImageProcessor", | |
| "do_normalize": true, | |
| "do_resize": true, | |
| "ensure_multiple_of": 1, | |
| "image_mean": [ | |
| 0.5, | |
| 0.5, | |
| 0.5 | |
| ], | |
| "image_std": [ | |
| 0.5, | |
| 0.5, | |
| 0.5 | |
| ], | |
| "keep_aspect_ratio": false, | |
| "resample": 2, | |
| "size": 224 | |
| } |