平展OpenCV/Numpy阵列

Question

我已经加载了PIL/OpenCV的RGB图像，并且我想将它的所有通道转换为单个1x（3 *宽*高）序列，以便将其馈送给ANN。我发现我可以简单地做：

rlist = [] 
glist = [] 
blist = [] 
for i in xrange(im.width): 
    for j in xrange(im.height): 
     r,g,b = im[i,j] 
     rlist.append(r) 
     glist.append(g) 
     blist.append(b) 
img_vec = rlist + blist + glist 

但显然这是非常低效率。有一些内部OpenCV/numpy例程有更快的方法吗？

Answer 1

作为一个简单的例子：

import Image 
import numpy as np 

im = Image.open('temp.png') 
data = np.array(im) 
flattened = data.flatten() 

print data.shape 
print flattened.shape 

这产生了：

(612, 812, 4) 
(1987776,) 

或者，而不是调用data.flatten()，你可以调用data.reshape(-1)。 -1被用作“找出给定维度应该是什么”的占位符。

请注意，这将产生一个向量（flattened）r0, g0, b0, r1, g1, b1, ... rn, gn, bn，而你想要一个向量r0, r1 ... rn, b0, b1, ... bn, g0, g1, ... gn。

为了得到你想要什么，只是叫

flattened = data.T.flatten() 

代替。