如何将图像多维数组数据拟合到python中的随机森林分类器中？

问题描述

我想使用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

我该如何解决这个问题？

解决方法

最基本的解决方案是将图像展平为单个行，每个像素都有一个元素。用最大的图像来做，用其他的垫来做。

但这是一个不好的解决方案，可能无法正常工作。

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