Python get_office Examples

Example #1

def get_data_loader(name, dataset_root, batch_size, train=True):
    """Get data loader by name."""
    if name == "mnist":
        return get_mnist(dataset_root, batch_size, train)
    elif name == "mnistm":
        return get_mnistm(dataset_root, batch_size, train)
    elif name == "svhn":
        return get_svhn(dataset_root, batch_size, train)
    elif name == "amazon31":
        return get_office(dataset_root, batch_size, 'amazon')
    elif name == "webcam31":
        return get_office(dataset_root, batch_size, 'webcam')
    elif name == "webcam10":
        return get_officecaltech(dataset_root, batch_size, 'webcam')

Example #2

File: utils.py Project: TangLisan/pytorch-office_finetune

def get_data_loader(name, dataset_root, batch_size, train=True):
    """Get data loader by name."""
    if name == "amazon31":
        return get_office(dataset_root, batch_size, 'amazon', train)
    elif name == "webcam31":
        return get_office(dataset_root, batch_size, 'webcam', train)

Example #3

File: utils.py Project: wogong/pytorch-office_finetune

def get_data_loader(name, dataset_root, batch_size, train=True):
    """Get data loader by name."""
    if name == "amazon31":
        return get_office(dataset_root, batch_size, 'amazon', train)
    elif name == "webcam31":
        return get_office(dataset_root, batch_size, 'webcam', train)