将旋转重新整形并应用于MNIST数据集

时间:2017-08-25 20:55:23

标签: python image-rotation mnist

我正在尝试旋转MNIST数据集中的图像(在csv中),另外的目标是将角度记录为标签(而不是数字0-9作为标签)。我有以下步骤:

sample_digit = df.iloc[42].values.reshape(28,28) # reshapes 784 => 28x28
rotated = rotate(sample_digit, angle) # rotates the reshaped data by 'angle' 

这适用于上面的图像。但我正在寻找的,似乎无法做到的是,将df循环到: 1)将每个数据点重塑为28x28, 2)应用随机旋转,和 3)记录角度像这样:

rotated_images = []
angles = []
for i, row in df.itertuples():
    reshaped = row.values.reshape(28,28)
    rotated = rotate(sample_digit, angle)
    rotated_images.append(rotated)
    angles.append(angle)

谢谢!

0 个答案:

没有答案