metadata
license: mit
tags:
- image-feature-extraction
Usage
import timm
from timm.data import resolve_data_config
from timm.data.transforms_factory import create_transform
from huggingface_hub import login
login() # login or use an access token
model = timm.create_model("hf-hub:kaiko-ai/midnight", pretrained=True)
transform = create_transform(**resolve_data_config(model.pretrained_cfg, model=model))
model.eval()