import random from PIL import Image import numpy as np class DataAugmentation: def __init__(self, dataset): self.dataset = dataset def rotate(self, image, angle): rotated_image = image.rotate(angle) return rotated_image def crop(self, image, crop_size): width, height = image.size left = random.randint(0, width - crop_size) upper = random.randint(0, height - crop_size) right = left + crop_size lower = upper + crop_size cropped_image = image.crop((left, upper, right, lower)) return cropped_image def mirror(self, image): mirrored_image = image.transpose(Image.FLIP_LEFT_RIGHT) return mirrored_image def augment(self, num_samples, crop_size): augmented_dataset = [] for i in range(num_samples): image = Image.open(self.dataset[i]) operations = [self.rotate, self.crop, self.mirror] operation = random.choice(operations) if operation == self.rotate: angle = random.randint(0, 360) augmented_image = self.rotate(image, angle) elif operation == self.crop: augmented_image = self.crop(image, crop_size) else: augmented_image = self.mirror(image) augmented_dataset.append(np.array(augmented_image)) return augmented_dataset
时间: 2024-02-14 14:26:17 浏览: 177
利用torch.utils.data.Dataset自定义数据加载类
3星 · 编辑精心推荐
这是一个数据增强的类,用于对数据集进行图像增强操作。它具有以下方法:
- `rotate(image, angle)`:旋转图像,接受一个图像和旋转角度作为参数,并返回旋转后的图像。
- `crop(image, crop_size)`:裁剪图像,接受一个图像和裁剪尺寸作为参数,并返回裁剪后的图像。
- `mirror(image)`:镜像图像,接受一个图像作为参数,并返回镜像后的图像。
- `augment(num_samples, crop_size)`:对数据集进行增强操作,接受增强样本数和裁剪尺寸作为参数,并返回增强后的数据集。
在 `augment` 方法中,随机选择一个操作(旋转、裁剪或镜像),然后根据选择的操作对图像进行相应的处理,并将增强后的图像添加到增强数据集中。
请问还有其他问题吗?
阅读全文