| # 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") | |