Как создать пакет изображений из одного изображения, добавив смещение?

Question

Я изучаю шумоподавление изображений и Pytorch. Я хочу получить серию изображений, сгенерированных из одного изображения. Например, у меня есть изображение, а затем случайным образом вырезать из него патч определенного размера. Затем я хочу добавить сдвиг на 1 или 2 пикселя, чтобы получить новое изображение с крошечной разницей. Что я мог сделать? Лучше использовать некоторые методы в PIL или другие?

Answer 1

Вы должны использовать transforms, чтобы сделать некоторое увеличение изображения для вашей проблемы.

Пока я читаю ваш комментарий, вы можете ограничить translate = (a, b) некоторыми небольшими случайными сдвигами в обоих измерениях.

import torchvision.transforms as transforms

transform = transforms.RandomAffine(degrees, translate=None, scale=None, shear=None, resample=False, fillcolor=0)

img = PIL.Image.open('path/img')

new_img = transform(img)

Если вы хотите выполнить больше преобразований, например Crop, сгруппируйте все transform в один большой transform, используя transforms.Compose. Вот ваша ссылка