rwightman / pytorch-image-models

PyTorch image models, scripts, pretrained weights

GitHub Docs

PyTorch Image Models

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.