Esempi in Python per get_imagenet_dataflow

Esempio n. 1

File: imagenet-resnet.py Progetto: 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')

Esempio n. 2

File: imagenet-resnet-se.py Progetto: 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)