Image Recognition Trainer

Trains a resnet50 base model to provide a confidence score of a supplied image across 7 classes.

Able to export prediction build models
Ability to run inference

• see readme for more details on folder structure

Python 3.12.4

• os
• timm
• torch
• torchvision --> datasets, transforms
• torch.utils.data --> DataLoader
• multiprocessing --> freeze_support
  *torch.optim.lr_scheduler --> StepLR