W&B + PyImageSearch MLOps Course

This repo contains the code for the course: https://pyimagesearch.mykajabi.com/offers/LQSsX59C

1. Download the dataset.

$ git clone -qq https://github.com/softwaremill/lemon-dataset.git
$ unzip -q lemon-dataset/data/lemon-dataset.zip

1.5 Open the params.py file and change the wandb parameters to your own project.

PROJECT_NAME = "wandb_course"
ENTITY = "pyimagesearch"  # just set this to your username

2. Run python prepare_data.py to prepare the dataset as an artifact and upload it to W&B.

3. Run python eda.py to do Exploratory Data Analysis on the dataset. We also upload the wandb Table from our analysis to W&B.

4. Train the model using python train.py

5. Evaluate the model using python eval.py