如何将图像(多维数组)数据放入python中的随机森林分类器中

Question

我想使用sklearn.ensemble 构建一个图像分类器

我有一个图像X_train的列表，其中

X_train[0].shape
Out[58]: (353, 1054, 3)

以及标量标签y_ train的列表。

每个图像X_train[i]具有不同的形状。当我试图将这些数据放入分类器时，我得到以下错误

clf=RandomForestClassifier(n_estimators=100)
clf.fit(X_train,y_train)
ValueError: setting an array element with a sequence

我该如何解决这个问题？

Answer 1

最基本的解决方案是将图像展平为一行，每个像素都有一个元素。按照最大图像的大小进行操作，并为其他图像填充。

不过，这是一个糟糕的解决方案，可能不会很好地工作。

图像分类需要更复杂的方法，超出了这个问题的范围。