license: mit | |
tags: | |
- image-feature-extraction | |
### Usage | |
```python | |
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() | |
``` |