INNER CODE UNIT · Python

get_dataset_raw_and_encoded

nv-tlabs/ATISS · scene_synthesis/datasets/__init__.py:47

def get_dataset_raw_and_encoded(
    config,
    filter_fn=lambda s: s,
    path_to_bounds=None,
    augmentations=None,
    split=["train", "val"]
):
    dataset = get_raw_dataset(config, filter_fn, path_to_bounds, split=split)
    encoding = dataset_encoding_factory(
        config.get("encoding_type"),
        dataset,
        augmentations,
        config.get("box_ordering", None)
    )

    return dataset, encoding

View source record →