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
评论列表
文章目录