我正在尝试旋转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)
谢谢!