如何将图像从(32,32,3)转换为(1,3072)?

时间:2018-12-17 07:30:52

标签: python scikit-learn

(关于图像分类)

将32x32x3图像转换为1x3072数组以使1024为红色,1024为绿色和1024为蓝色的最简单方法是什么?

2 个答案:

答案 0 :(得分:2)

img = img.transpose(2, 0, 1)  # Depends on your original order
img = img.reshape(1, -1)

请参阅:

答案 1 :(得分:2)

使用numpy:(arr是图像数组)

swapped = np.swapaxes(arr,0,2)
flattened = swapped.flatten()

您的数组将位于flattened

在第一个轴和最后一个轴之间切换,然后展平结果。