datahandler.py 文件源码

def process_data(augmentation=2):
    cifar = load_data()
    cifar['train']['x'], m, sd = normalize(cifar['train']['x'])
    cifar['test']['x'], m, sd = normalize(cifar['test']['x'], M=m, Sd=sd)
    if augmentation > 0:
        cifar['train']['x'], cifar['train']['y'] = pad_rightleft(cifar['train']['x'], cifar['train']['y'],
                                                                 mixratio=augmentation)
    if augmentation > 1.0:
        cifar['train']['x'], cifar['train']['y'] = pad_addnoise(cifar['train']['x'], cifar['train']['y'],
                                                                mixratio=augmentation - 1.0)
#    data.save_pkl(cifar, savename='cifar_processed.pkl')
    return cifar