Unverified Commit 1b8f3566 authored by Nicolas Hug's avatar Nicolas Hug Committed by GitHub
Browse files

Change expected directory from "Kitti" to "KittiFlow" for KittiFlow dataset (#4889)

parent a4ca717f
......@@ -1945,7 +1945,7 @@ class KittiFlowTestCase(datasets_utils.ImageDatasetTestCase):
FEATURE_TYPES = (PIL.Image.Image, PIL.Image.Image, (np.ndarray, type(None)), (np.ndarray, type(None)))
def inject_fake_data(self, tmpdir, config):
root = pathlib.Path(tmpdir) / "Kitti"
root = pathlib.Path(tmpdir) / "KittiFlow"
num_examples = 2 if config["split"] == "train" else 3
for split_dir in ("training", "testing"):
......
......@@ -155,7 +155,7 @@ class KittiFlow(FlowDataset):
The dataset is expected to have the following structure: ::
root
Kitti
KittiFlow
testing
image_2
training
......@@ -176,7 +176,7 @@ class KittiFlow(FlowDataset):
verify_str_arg(split, "split", valid_values=("train", "test"))
root = Path(root) / "Kitti" / (split + "ing")
root = Path(root) / "KittiFlow" / (split + "ing")
images1 = sorted(glob(str(root / "image_2" / "*_10.png")))
images2 = sorted(glob(str(root / "image_2" / "*_11.png")))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment