def __getitem__(self, idx):
    try:
        (im0_path, im1_path, raftflow_path) = self.dataset[self.split][idx]
        raftflow = frame_utils.readFlow(raftflow_path)
    except:
        (im0_path, im1_path) = self.dataset[self.split][idx]
        raftflow = np.array([])
    img0 = skimage.io.imread(im0_path)
    img1 = skimage.io.imread(im1_path)
    img0 = torch.tensor((img0 / 255.0)).float()
    img1 = torch.tensor((img1 / 255.0)).float()
    return (img0, img1, np.array([]), np.array([]), [im0_path, im1_path], raftflow)
def __getitem__(self, idx):
    try:
        (im0_path, im1_path, raftflow_path) = self.dataset[self.split][idx]
        raftflow = frame_utils.readFlow(raftflow_path)
    except:
        (im0_path, im1_path) = self.dataset[self.split][idx]
        raftflow = np.array([])
    img0 = skimage.io.imread(im0_path)
    img1 = skimage.io.imread(im1_path)
    img0 = torch.tensor((img0 / 255.0)).float()
    img1 = torch.tensor((img1 / 255.0)).float()
    return (img0, img1, np.array([]), np.array([]), [im0_path, im1_path], raftflow)