nip.image_classification.pretrained_models#
Classes for loading pretrained image models, to use their embeddings.
Models are loaded from a ‘hub’ somewhere (e.g. PyTorch Image Models (timm) or Hugging Face Transformers).
The models are then used to extract embeddings from images, which can be used to aid agents in image classification tasks.
Classes
|
Base class for pretrained image models. |
|
Resnet18 model trained on CIFAR-10. |
|
Base class for pretrained ResNet models using PyTorch Image Models (timm). |