def METHOD_NAME(self, img0, img1, timestep=0.5, scale=1.0):
    imgs = torch.cat((img0, img1), 1)
    scale_list = [8/scale, 4/scale, 2/scale, 1/scale]
    flow, mask, merged = self.flownet(imgs, timestep, scale_list)
    return merged[3]