flatten用于展平数组。 首先,假设你有一个灰度图像。 这个图像中只有1到9的3x3像素的点。 执行flatten操作。 首先每行分开,在第二行中跟在第一行的后面,组成新的数组1、2、3、4、5、6,最后在新组成第三行的7、8、9的数组后面组成最终的数组。
接下来验证一下我们的预想吧。 首先导入numpy包,用它创建3x3的二维数组a,然后打印a。
灰色图像是一维的,相对于其flatten :
经过flatten操作,可以看出本来3行的排列变成了1行。
如果是彩色图像,则有三个颜色通道。 进行fltten时的步骤也相同。 将每个通道展平,将第二个通道连接到第一个通道的后面,将第三个通道连接到第二个通道的后面。