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

Before Change

def make_dataset(dir, split, dataset_type="clean"):
    training_dir = os.path.join(dir,"training")
    flow_dir = "flow"
    assert(os.path.isdir(os.path.join(training_dir,flow_dir)))
    img_dir = dataset_type
    assert(os.path.isdir(os.path.join(training_dir,img_dir)))

After Change

        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