Python get_imagenet_dataflowの例

コード例 #1

ファイル: imagenet-resnet.py プロジェクト: 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')

コード例 #2

ファイル: imagenet-resnet-se.py プロジェクト: 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)