INNER CODE UNIT · Python

dataset

lufficc/SSD · ssd/data/datasets/__init__.py:26

        dataset = factory(**args)
        datasets.append(dataset)
    # for testing, return a list of datasets
    if not is_train:
        return datasets
    dataset = datasets[0]
    if len(datasets) > 1:
        dataset = ConcatDataset(datasets)

    return [dataset]

View source record →