SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix SAM example in documentation (#22887)

Browse Source 
fix sam example
This commit is contained in:
fxmarty
2023-04-20 12:22:42 +02:00
committed by GitHub
parent cb47293eba
commit 4cfe328bae
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
docs/source/en/model_doc/sam.mdx
View File

@@ -39,12 +39,14 @@ The original code can be found [here](https://github.com/facebookresearch/segmen
Below is an example on how to run mask generation given an image and a 2D point: Below is an example on how to run mask generation given an image and a 2D point:
```python ```python
import torch
from PIL import Image from PIL import Image
import requests import requests
from transformers import SamModelForMaskedGeneration, SamProcessor from transformers import SamModel, SamProcessor
model = SamModelForMaskedGeneration.from_pretrained("facebook/sam-vit-huge") device = "cuda" if torch.cuda.is_available() else "cpu"
processsor = SamProcessor.from_pretrained("facebook/sam-vit-huge") model = SamModel.from_pretrained("facebook/sam-vit-huge").to(device)
processor = SamProcessor.from_pretrained("facebook/sam-vit-huge")
img_url = "https://huggingface.co/ybelkada/segment-anything/resolve/main/assets/car.png" img_url = "https://huggingface.co/ybelkada/segment-anything/resolve/main/assets/car.png"
raw_image = Image.open(requests.get(img_url, stream=True).raw).convert("RGB") raw_image = Image.open(requests.get(img_url, stream=True).raw).convert("RGB")