8753096898d9b23e0ce05a1808e4b93f7475b051,datasets/mpisintel.py,,make_dataset,#,15

Before Change

        root_filename = flow_map[:-8]
        frame_nb = int(flow_map[-8:-4])
        img1 = os.path.join(img_dir,root_filename+str(frame_nb).zfill(4)+".png")
        img2 = os.path.join(img_dir,root_filename+str(frame_nb+1).zfill(4)+".png")
        flow_map = os.path.join(flow_dir,flow_map)
        if not (os.path.isfile(os.path.join(dir,img1)) or os.path.isfile(os.path.join(dir,img2))):
            continue

After Change

        flow_map = os.path.relpath(flow_map,os.path.join(dataset_dir,flow_dir))

        scene_dir, filename = os.path.split(flow_map)
        no_ext_filename = os.path.splitext(filename)[0]
        prefix, frame_nb = no_ext_filename.split("_")
        frame_nb = int(frame_nb)
        img1 = os.path.join(img_dir, scene_dir, "{}_{:04d}.png".format(prefix, frame_nb))
        img2 = os.path.join(img_dir, scene_dir, "{}_{:04d}.png".format(prefix, frame_nb + 1))
        flow_map = os.path.join(flow_dir,flow_map)

Instances