Python cifar10_loader示例

示例#1

文件： check_selection.py 项目： hhjung1202/SelGN

def dataset_selector(data):
    if data == 'mnist':
        return dataset.MNIST_loader(img_size=args.img_size)
    elif data == 'svhn':
        return dataset.SVHN_loader(img_size=32)
    elif data == "usps":
        return dataset.usps_loader(img_size=args.img_size)
    elif data == "mnistm":
        return dataset.MNIST_M_loader(img_size=args.img_size)
    elif data == "cifar10":
        return dataset.cifar10_loader(args)

示例#2

def dataset_selector(db):
    if db == "cifar10":
        return dataset.cifar10_loader(args)
    else:
        return dataset.cifar100_loader(args)

示例#3

def dataset_selector():
    # return dataset.cifar100_loader(args)
    return dataset.cifar10_loader(args)