Я хочу преобразовать изображение в 2D-массив с 5 столбцами, где каждая строка имеет форму [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