def dist(a, b):
    return sum((i-j)**2 for i, j in zip(a, b))
def __init__(self, root, transforms=None):

        super().__init__(root=root)
        self.transforms = transforms

        self._flow_list = []
        self._image_list = []