将数组重塑为rgb矩阵

时间:2016-08-04 18:36:34

标签: python arrays numpy

我已经在numpy数组中加载了100x100 rgb图像。然后我将它转换为30000x1 numpy数组,以通过机器学习模型。该模型的输出也是一个30000x1的numpy阵列。如何将此数组转换回100x100 3元组的numpy数组,以便打印生成的rgb图像?

如果初始数组为k=2,则会将其展开到[r1 g1 b1],[r2 g2 b2],...,[]。我需要以[r1 g1 b1 r2 g2 b2 ...]的形式回复它。

我用来将图像加载为数组:

[r1 g1 b1],[r2 g2 b2],...,[]

我试过.reshape((100,100,3))并且我得到了黑色输出图像。机器学习模型是正确的,它不是获得黑色输出的原因。

1 个答案:

答案 0 :(得分:1)

尝试NOTHING

reshape((3, 100, 100))