Update README.md
Browse files
README.md
CHANGED
|
@@ -54,8 +54,8 @@ from PIL import Image
|
|
| 54 |
import requests
|
| 55 |
from transformers import SamHQModel, SamHQProcessor
|
| 56 |
|
| 57 |
-
model = SamHQModel.from_pretrained("
|
| 58 |
-
processor = SamHQProcessor.from_pretrained("
|
| 59 |
|
| 60 |
img_url = "https://raw.githubusercontent.com/SysCV/sam-hq/refs/heads/main/demo/input_imgs/example1.png"
|
| 61 |
raw_image = Image.open(requests.get(img_url, stream=True).raw).convert("RGB")
|
|
@@ -79,7 +79,7 @@ The pipeline is made for automatic mask generation. The following snippet demons
|
|
| 79 |
|
| 80 |
```python
|
| 81 |
from transformers import pipeline
|
| 82 |
-
generator = pipeline("mask-generation", model="
|
| 83 |
image_url = "https://raw.githubusercontent.com/SysCV/sam-hq/refs/heads/main/demo/input_imgs/example1.png"
|
| 84 |
outputs = generator(image_url, points_per_batch=256)
|
| 85 |
```
|
|
@@ -210,8 +210,8 @@ import torch
|
|
| 210 |
from transformers import SamHQModel, SamHQProcessor
|
| 211 |
|
| 212 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 213 |
-
model = SamHQModel.from_pretrained("
|
| 214 |
-
processor = SamHQProcessor.from_pretrained("
|
| 215 |
|
| 216 |
from PIL import Image
|
| 217 |
import requests
|
|
|
|
| 54 |
import requests
|
| 55 |
from transformers import SamHQModel, SamHQProcessor
|
| 56 |
|
| 57 |
+
model = SamHQModel.from_pretrained("syscv-community/sam-hq-vit-largel")
|
| 58 |
+
processor = SamHQProcessor.from_pretrained("syscv-community/sam-hq-vit-large")
|
| 59 |
|
| 60 |
img_url = "https://raw.githubusercontent.com/SysCV/sam-hq/refs/heads/main/demo/input_imgs/example1.png"
|
| 61 |
raw_image = Image.open(requests.get(img_url, stream=True).raw).convert("RGB")
|
|
|
|
| 79 |
|
| 80 |
```python
|
| 81 |
from transformers import pipeline
|
| 82 |
+
generator = pipeline("mask-generation", model="syscv-community/sam-hq-vit-large", device=0, points_per_batch=256)
|
| 83 |
image_url = "https://raw.githubusercontent.com/SysCV/sam-hq/refs/heads/main/demo/input_imgs/example1.png"
|
| 84 |
outputs = generator(image_url, points_per_batch=256)
|
| 85 |
```
|
|
|
|
| 210 |
from transformers import SamHQModel, SamHQProcessor
|
| 211 |
|
| 212 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 213 |
+
model = SamHQModel.from_pretrained("syscv-community/sam-hq-vit-large").to(device)
|
| 214 |
+
processor = SamHQProcessor.from_pretrained("syscv-community/sam-hq-vit-large")
|
| 215 |
|
| 216 |
from PIL import Image
|
| 217 |
import requests
|