Total runs:
53.8K
Run Growth:
12.3K
Growth Rate:
22.71%
Sponsored by
iMean AI Trip Planner
smp-hub / upernet-swin-large
Model's Last Updated: April 13 2025
image-segmentation
Introduction of upernet-swin-large
Model Details of upernet-swin-large
UPerNet Model Card
Table of Contents:
Load trained model
- Install requirements.
pip install -U segmentation_models_pytorch albumentations
- Run inference.
import torch
import requests
import numpy as np
import albumentations as A
import segmentation_models_pytorch as smp
from PIL import Image
device = "cuda" if torch.cuda.is_available() else "cpu"
# Load pretrained model and preprocessing function
checkpoint = "smp-hub/upernet-swin-large"
model = smp.from_pretrained(checkpoint).eval().to(device)
preprocessing = A.Compose.from_pretrained(checkpoint)
# Load image
url = "https://huggingface.co/datasets/hf-internal-testing/fixtures_ade20k/resolve/main/ADE_val_00000001.jpg"
image = Image.open(requests.get(url, stream=True).raw)
# Preprocess image
np_image = np.array(image)
normalized_image = preprocessing(image=np_image)["image"]
input_tensor = torch.as_tensor(normalized_image)
input_tensor = input_tensor.permute(2, 0, 1).unsqueeze(0) # HWC -> BCHW
input_tensor = input_tensor.to(device)
# Perform inference
with torch.no_grad():
output_mask = model(input_tensor)
# Postprocess mask
mask = mask.argmax(1).cpu().numpy() # argmax over predicted classes (channels dim)
Model init parameters
model_init_params = {
"encoder_name": "tu-swin_large_patch4_window12_384",
"encoder_depth": 5,
"encoder_weights": None,
"decoder_channels": 512,
"decoder_use_norm": "batchnorm",
"in_channels": 3,
"classes": 150,
"activation": None,
"upsampling": 4,
"aux_params": None,
"img_size": 512
}
Dataset
Dataset name: ADE20K
More Information
- Library: https://github.com/qubvel/segmentation_models.pytorch
- Docs: https://smp.readthedocs.io/en/latest/
This model has been pushed to the Hub using the PytorchModelHubMixin
Runs of smp-hub upernet-swin-large on huggingface.co
221
Total runs
0
24-hour runs
3
3-day runs
31
7-day runs
208
30-day runs
More Information About upernet-swin-large huggingface.co Model
Url of upernet-swin-large
upernet-swin-large huggingface.co Url
Provider of upernet-swin-large huggingface.co
Other API from smp-hub
Updated:January 16 2025
Total runs:
44.4K
Run Growth:
-6.7K
Growth Rate:
-14.89%
Total runs:
42.7K
Run Growth:
4.7K
Growth Rate:
11.07%
Total runs:
26.7K
Run Growth:
1.6K
Growth Rate:
5.87%
Total runs:
18.0K
Run Growth:
-118.8K
Growth Rate:
-667.84%
Total runs:
6.2K
Run Growth:
716
Growth Rate:
11.56%
Total runs:
4.6K
Run Growth:
681
Growth Rate:
14.66%
Total runs:
4.4K
Run Growth:
87
Growth Rate:
1.91%
Total runs:
4.2K
Run Growth:
632
Growth Rate:
14.93%
Total runs:
3.9K
Run Growth:
3.2K
Growth Rate:
84.35%
Total runs:
3.3K
Run Growth:
-3.3K
Growth Rate:
-96.52%
Total runs:
2.4K
Run Growth:
1.1K
Growth Rate:
45.68%
Total runs:
2.2K
Run Growth:
137
Growth Rate:
6.26%
Total runs:
1.7K
Run Growth:
775
Growth Rate:
50.23%
Total runs:
1.6K
Run Growth:
332
Growth Rate:
19.77%
Total runs:
1.6K
Run Growth:
-52
Growth Rate:
-3.20%
Total runs:
1.3K
Run Growth:
218
Growth Rate:
16.87%
Total runs:
1.3K
Run Growth:
262
Growth Rate:
20.14%
Total runs:
1.3K
Run Growth:
469
Growth Rate:
35.50%
Total runs:
1.1K
Run Growth:
451
Growth Rate:
42.71%
Total runs:
1.1K
Run Growth:
293
Growth Rate:
25.37%
Total runs:
983
Run Growth:
355
Growth Rate:
35.25%
Total runs:
890
Run Growth:
17
Growth Rate:
1.68%
Total runs:
819
Run Growth:
200
Growth Rate:
24.78%
Total runs:
752
Run Growth:
651
Growth Rate:
85.88%
Total runs:
707
Run Growth:
-517
Growth Rate:
-76.71%
Total runs:
612
Run Growth:
151
Growth Rate:
26.40%
Total runs:
520
Run Growth:
143
Growth Rate:
27.29%
Total runs:
514
Run Growth:
-315
Growth Rate:
-50.97%
Total runs:
335
Run Growth:
236
Growth Rate:
75.16%
Total runs:
300
Run Growth:
76
Growth Rate:
25.68%
Total runs:
249
Run Growth:
90
Growth Rate:
32.03%
Total runs:
215
Run Growth:
87
Growth Rate:
40.85%
Total runs:
186
Run Growth:
92
Growth Rate:
51.98%
Total runs:
173
Run Growth:
139
Growth Rate:
80.35%
Total runs:
146
Run Growth:
-324
Growth Rate:
-225.00%
Total runs:
146
Run Growth:
-12
Growth Rate:
-8.39%
Total runs:
144
Run Growth:
111
Growth Rate:
77.08%
Total runs:
136
Run Growth:
120
Growth Rate:
75.95%
Total runs:
131
Run Growth:
-24
Growth Rate:
-17.65%
Total runs:
130
Run Growth:
52
Growth Rate:
40.31%
Total runs:
123
Run Growth:
-115
Growth Rate:
-96.64%
Total runs:
122
Run Growth:
114
Growth Rate:
93.44%
Total runs:
113
Run Growth:
34
Growth Rate:
29.82%
Total runs:
100
Run Growth:
78
Growth Rate:
80.41%
Total runs:
94
Run Growth:
-99
Growth Rate:
-92.52%
Total runs:
92
Run Growth:
45
Growth Rate:
52.94%
Total runs:
74
Run Growth:
43
Growth Rate:
50.00%
Total runs:
65
Run Growth:
55
Growth Rate:
84.62%
Total runs:
64
Run Growth:
55
Growth Rate:
85.94%
Total runs:
62
Run Growth:
22
Growth Rate:
33.33%
Total runs:
60
Run Growth:
12
Growth Rate:
19.67%
Total runs:
56
Run Growth:
32
Growth Rate:
58.18%
Total runs:
48
Run Growth:
5
Growth Rate:
10.20%
Total runs:
48
Run Growth:
7
Growth Rate:
14.89%
Total runs:
44
Run Growth:
3
Growth Rate:
6.82%
Total runs:
42
Run Growth:
32
Growth Rate:
76.19%
Total runs:
42
Run Growth:
-1
Growth Rate:
-2.56%
Total runs:
37
Run Growth:
11
Growth Rate:
31.43%
Total runs:
33
Run Growth:
24
Growth Rate:
72.73%
Total runs:
30
Run Growth:
-7
Growth Rate:
-22.58%
Total runs:
27
Run Growth:
27
Growth Rate:
79.41%
Total runs:
24
Run Growth:
7
Growth Rate:
36.84%
Total runs:
24
Run Growth:
18
Growth Rate:
75.00%
Total runs:
24
Run Growth:
1
Growth Rate:
4.00%
Total runs:
24
Run Growth:
18
Growth Rate:
69.23%
Total runs:
24
Run Growth:
4
Growth Rate:
16.00%
Total runs:
19
Run Growth:
13
Growth Rate:
65.00%
Total runs:
19
Run Growth:
6
Growth Rate:
31.58%
Total runs:
19
Run Growth:
4
Growth Rate:
21.05%
Total runs:
19
Run Growth:
14
Growth Rate:
73.68%
Total runs:
18
Run Growth:
3
Growth Rate:
15.79%
Total runs:
17
Run Growth:
11
Growth Rate:
57.89%
Total runs:
17
Run Growth:
0
Growth Rate:
0.00%
Total runs:
17
Run Growth:
2
Growth Rate:
11.76%
Total runs:
16
Run Growth:
8
Growth Rate:
50.00%
Total runs:
15
Run Growth:
7
Growth Rate:
46.67%
Total runs:
15
Run Growth:
8
Growth Rate:
57.14%
Total runs:
15
Run Growth:
8
Growth Rate:
53.33%
Total runs:
14
Run Growth:
-1
Growth Rate:
-7.14%
Total runs:
14
Run Growth:
-1
Growth Rate:
-7.69%