6c3a117b9b39fd7f993c85091f7a1a0d96d10d08,datasets/KITTI.py,,load_flow_from_png,#Any#,16

Before Change

def load_flow_from_png(png_path):
    return(imread(png_path)[:,:,0:2].astype(float) - 128)


def make_dataset(dir, split, occ=True):

After Change

def load_flow_from_png(png_path):
    flo_file = cv2.imread(png_path,cv2.IMREAD_UNCHANGED)
    flo_img = flo_file[:,:,1::-1].astype(np.float32)
    invalid = (flo_file[:,:,2] == 0)
    flo_img = flo_img - 32768
    flo_img = flo_img / 64
    flo_img[invalid, :] = np.NaN
    return(flo_img)

Instances