transformers SegformerForSemanticSegmentation image-segmentation custom_code region:us
---
library_name: transformers
pipeline_tag: image-segmentation
---

# How to use 
either load the model 
```python
from transformers import AutoModelForImageSegmentation
model = AutoModelForImageSegmentation.from_pretrained("briaai/RMBG-1.4",revision ="refs/pr/9",trust_remote_code=True)
```
or load the pipeline
```python
from transformers import pipeline

pipe = pipeline("image-segmentation", model="briaai/RMBG-1.4",revision ="refs/pr/9", trust_remote_code=True)

numpy_mask = pipe("img_path") # outputs numpy mask

pipe("image_path",out_name="myout.png") # applies mask and saves the extracted image as `myout.png`

```




# parameters : 
for the pipeline you can use the following parameters : 
* `model_input_size` : default to [1024,1024]
* `out_name` : if specified it will use the numpy mask to extract the image and save it using the `out_name`
* `preprocess_image` : original method created by briaai
* `postprocess_image` : original method created by briaai

# disclamer 
I do not own, distribute or take credit for this model. 

All rights belong to [briaai](https://huggingface.co/briaai/) 

This repo is a temporary one to test out the custom architecture for [RMBG-1.4](https://huggingface.co/briaai/RMBG-1.4), please do refer to the original model.
