augment.py

# Cesar Garcia

# Imports
import numpy as np
from PIL import Image
from datasets import load_dataset

# Load dataset from Hugging Face
dataset = load_dataset("gcesar/spinach", download_mode="force_redownload")

# For loop to traverse dataset
for i in range(20):

    dict_row = dataset['train'][i]
    img = dict_row.get('image')

    # Flip image left to right
    img_lr = img.transpose(Image.FLIP_LEFT_RIGHT)

    # Rotate image
    img_rot = img_lr.rotate(5, resample=Image.BILINEAR)

    # Save images
    img_rot.save(f"img_aug{i}.jpg")