Papers with Code - PyTorch Image Models (original) (raw)
PyTorch Image Models (TIMM) is a library for state-of-the-art image classification. With this library you can:
- Choose from 300+ pre-trained state-of-the-art image classification models.
- Train models afresh on research datasets such as ImageNet using provided scripts.
- Finetune pre-trained models on your own datasets, including the latest cutting edge models.