我想将图像转换为包含5列的2D数组,其中每行的格式为[r, g, b, x, y]
。 x,y是像素的位置,r,g,b是像素值。 (我将使用此数组作为机器学习模型的输入)。在python中有比这更有效的实现吗?
import Image
import numpy as np
im = Image.open("farm.jpg")
col,row = im.size
data = np.zeros((row*col, 5))
pixels = im.load()
for i in range(row):
for j in range(col):
r,g,b = pixels[i,j]
data[i*col + j,:] = r,g,b,i,j
答案 0 :(得分:14)
我最近必须写这篇文章,最后以
结束indices = np.dstack(np.indices(im.shape[:2]))
data = np.concatenate((im, indices), axis=-1)
im
是一个numpy数组。你可能最好用
from scipy.misc import imread
im = imread("farm.jpg")
或者,如果您安装了Scikit Image,那就更好了
from skimage.io import imread
im = imread("farm.jpg")
答案 1 :(得分:5)
我不确定这是否非常有效。但是你走了,说arr = np.array(im)
;然后你可以做这样的事情。
>>> arr = np.arange(150).reshape(5, 10, 3)
>>> x, y, z = arr.shape
>>> indices = np.vstack(np.unravel_index(np.arange(x*y), (y, x))).T
#or indices = np.hstack((np.repeat(np.arange(y), x)[:,np.newaxis], np.tile(np.arange(x), y)[:,np.newaxis]))
>>> np.hstack((arr.reshape(x*y, z), indices))
array([[ 0, 1, 2, 0, 0],
[ 3, 4, 5, 0, 1],
[ 6, 7, 8, 0, 2],
[ 9, 10, 11, 0, 3],
[ 12, 13, 14, 0, 4],
[ 15, 16, 17, 1, 0],
[ 18, 19, 20, 1, 1],
[ 21, 22, 23, 1, 2],
[ 24, 25, 26, 1, 3],
[ 27, 28, 29, 1, 4],
[ 30, 31, 32, 2, 0],
[ 33, 34, 35, 2, 1],
[ 36, 37, 38, 2, 2],
...
[129, 130, 131, 8, 3],
[132, 133, 134, 8, 4],
[135, 136, 137, 9, 0],
[138, 139, 140, 9, 1],
[141, 142, 143, 9, 2],
[144, 145, 146, 9, 3],
[147, 148, 149, 9, 4]])
答案 2 :(得分:1)
我用过" +"组合两个元组,并使用.append()
制作"数据" list.No需要在这里使用Numpy。
row,col = im.size
data=[] #r,g,b,i,j
pixels=im.load()
for i in range(row):
for j in range(col):
data.append(pixels[i,j]+(i,j))