Python get_imagenet_dataflow Exemples

Exemple #1

Fichier : imagenet-resnet.py Projet : NoobFang/tensorpack

def get_data(name):
    isTrain = name == 'train'
    augmentors = fbresnet_augmentor(isTrain)
    datadir = args.data
    return get_imagenet_dataflow(datadir,
                                 name,
                                 BATCH_SIZE,
                                 augmentors,
                                 dir_structure='original')

Exemple #2

Fichier : imagenet-resnet-se.py Projet : ycyoon/tensorpack

def get_data(name, batch):
    isTrain = name == 'train'
    augmentors = fbresnet_augmentor(isTrain)
    datadir = args.data
    return get_imagenet_dataflow(datadir, name, batch, augmentors)