Method __init__

diff2flow/dataset/image_preprocessing.py:7–9

(self, size)

class CenterCropResize(nn.Module):
    def __init__(self, size):
        super().__init__()
        self.size = size

    def forward(self, x):
        h, w = x.shape[-2:]