data_helpers.py 文件源码

def augment_test_data(images):
    more_images = np.zeros((4*images.shape[0], images.shape[1], images.shape[2],))
    for i in range(images.shape[0]):
        for j in range(4):
            rotation = j*90
            more_images[4*i+j] = ndimage.rotate(images[i], rotation)
    return more_images