在图像上执行SVM

Question

from scipy.misc import imread
from sklearn import svm

X_train = []
y_train = []

def f(dir, value):
    files = listdir(dir)
    for file in files:
        image_data = imread(dir + file)
        X_train.append(image_data)
        y_train.append(value)


f('data1/', 0)
f('data2/', 1)

model = svm.SVC().fit(X_train, y_train)

我尝试使用SVM来预测给定的图片是属于data1还是data2。

当我运行上面的代码时，我收到错误ValueError: setting an array element with a sequence.这是因为图像尺寸不同吗？如果是这样，我怎么能从图像中心裁剪50像素？