from urllib.request import urlopen
from PIL import Image
import timm
img = Image.open(urlopen(
'https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/beignets-task-guide.png'
))
model = timm.create_model('hrnet_w18.ms_aug_in1k', pretrained=True)
model = model.eval()
# get model specific transforms (normalization, resize)
data_config = timm.data.resolve_model_data_config(model)
transforms = timm.data.create_transform(**data_config, is_training=False)
output = model(transforms(img).unsqueeze(0)) # unsqueeze single image into batch of 1
top5_probabilities, top5_class_indices = torch.topk(output.softmax(dim=1) * 100, k=5)
Feature Map Extraction
from urllib.request import urlopen
from PIL import Image
import timm
img = Image.open(urlopen(
'https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/beignets-task-guide.png'
))
model = timm.create_model(
'hrnet_w18.ms_aug_in1k',
pretrained=True,
features_only=True,
)
model = model.eval()
# get model specific transforms (normalization, resize)
data_config = timm.data.resolve_model_data_config(model)
transforms = timm.data.create_transform(**data_config, is_training=False)
output = model(transforms(img).unsqueeze(0)) # unsqueeze single image into batch of 1for o in output:
# print shape of each feature map in output# e.g.:# torch.Size([1, 64, 112, 112])# torch.Size([1, 128, 56, 56])# torch.Size([1, 256, 28, 28])# torch.Size([1, 512, 14, 14])# torch.Size([1, 1024, 7, 7])print(o.shape)
Image Embeddings
from urllib.request import urlopen
from PIL import Image
import timm
img = Image.open(urlopen(
'https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/beignets-task-guide.png'
))
model = timm.create_model(
'hrnet_w18.ms_aug_in1k',
pretrained=True,
num_classes=0, # remove classifier nn.Linear
)
model = model.eval()
# get model specific transforms (normalization, resize)
data_config = timm.data.resolve_model_data_config(model)
transforms = timm.data.create_transform(**data_config, is_training=False)
output = model(transforms(img).unsqueeze(0)) # output is (batch_size, num_features) shaped tensor# or equivalently (without needing to set num_classes=0)
output = model.forward_features(transforms(img).unsqueeze(0))
# output is unpooled, a (1, 2048, 7, 7) shaped tensor
output = model.forward_head(output, pre_logits=True)
# output is a (1, num_features) shaped tensor
Model Comparison
Explore the dataset and runtime metrics of this model in timm
model results
.
Citation
@article{WangSCJDZLMTWLX19,
title={Deep High-Resolution Representation Learning for Visual Recognition},
author={Jingdong Wang and Ke Sun and Tianheng Cheng and
Borui Jiang and Chaorui Deng and Yang Zhao and Dong Liu and Yadong Mu and
Mingkui Tan and Xinggang Wang and Wenyu Liu and Bin Xiao},
journal = {TPAMI}
year={2019}
}
Runs of timm hrnet_w18.ms_aug_in1k on huggingface.co
24.5K
Total runs
0
24-hour runs
1.4K
3-day runs
-720
7-day runs
4.8K
30-day runs
More Information About hrnet_w18.ms_aug_in1k huggingface.co Model
hrnet_w18.ms_aug_in1k huggingface.co is an AI model on huggingface.co that provides hrnet_w18.ms_aug_in1k's model effect (), which can be used instantly with this timm hrnet_w18.ms_aug_in1k model. huggingface.co supports a free trial of the hrnet_w18.ms_aug_in1k model, and also provides paid use of the hrnet_w18.ms_aug_in1k. Support call hrnet_w18.ms_aug_in1k model through api, including Node.js, Python, http.
hrnet_w18.ms_aug_in1k huggingface.co is an online trial and call api platform, which integrates hrnet_w18.ms_aug_in1k's modeling effects, including api services, and provides a free online trial of hrnet_w18.ms_aug_in1k, you can try hrnet_w18.ms_aug_in1k online for free by clicking the link below.
timm hrnet_w18.ms_aug_in1k online free url in huggingface.co:
hrnet_w18.ms_aug_in1k is an open source model from GitHub that offers a free installation service, and any user can find hrnet_w18.ms_aug_in1k on GitHub to install. At the same time, huggingface.co provides the effect of hrnet_w18.ms_aug_in1k install, users can directly use hrnet_w18.ms_aug_in1k installed effect in huggingface.co for debugging and trial. It also supports api for free installation.
hrnet_w18.ms_aug_in1k install url in huggingface.co: