Model card

A Swin-v2 based species classification model. Original model: timm/swinv2_tiny_window16_256.ms_in1k

Model Details

Model Type: Animal classification
Model Stats:
Image size: 256 x 256

Model Usage

Image Embeddings


import timm
import torch
import torchvision.transforms as T

from PIL import Image
from urllib.request import urlopen

model = timm.create_model("hf-hub:cermakvo/wildintel-species-classifier-swin-t", pretrained=True)
model = model.eval()

transforms = T.Compose([T.Resize(256), 
                  T.ToTensor(), 
                  T.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])]) 

img = Image.open(urlopen(
    'https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/beignets-task-guide.png'
))

output = model(transforms(img).unsqueeze(0))  # output is (batch_size, num_features) shaped tensor
# output is a (1, num_features) shaped tensor