6c3a117b9b39fd7f993c85091f7a1a0d96d10d08,datasets/KITTI.py,,KITTI_loader,#,43

Before Change

def KITTI_loader(root,path_imgs, path_flo):
    imgs = [os.path.join(root,path) for path in path_imgs]
    flo = os.path.join(root,path_flo)
    return [imread(img) for img in imgs],load_flow_from_png(flo)


def KITTI_occ(root, transform=None, target_transform=None,

After Change

def KITTI_loader(root,path_imgs, path_flo):
    imgs = [os.path.join(root,path) for path in path_imgs]
    flo = os.path.join(root,path_flo)
    return [cv2.imread(img)[:,:,::-1].astype(np.float32) for img in imgs],load_flow_from_png(flo)


def KITTI_occ(root, transform=None, target_transform=None,

Instances