Image Classification

COMP 3340 Coursework

🔧 Skills Involved

Data preprocessing

Data transform and data augmentation

dataset.py

Dataset and Data loader

datamodule.py

Models

pretrain.py

pre-trained models and finetune

Visualization

Image sample

Training results demo in Wandb

Test accuracy