INNER CODE UNIT · Python

build_dataset_func

OpenMeshLab/MeshXL · main.py:104

def build_dataset_func(args):
    
    datasets = {
        'train': [], 
        'test': []
    }
    
    for dataset in args.dataset.split(','):
        dataset_module = importlib.import_module(f'datasets.{dataset}')
        datasets['train'].append(
            dataset_module.Dataset(args, split_set="train", augment=args.augment)
        )
        datasets['test'].append(
            dataset_module.Dataset(args, split_set="val", augment=False)
        )
    datasets['train'] = torch.utils.data.ConcatDataset(datasets['train'])
    
    train_sampler, train_loader = build_dataloader_func(args, datasets['train'], split='train')

View source record →