import torch


class ImageTransform:
    def __init__(self, parameters):
        super(ImageTransform, self).__init__()
        self.params = parameters

    def __call__(self, sample):
        image = sample.squeeze(0)
        return image
