在训练 SVM 对图像进行分类时设置具有序列错误的数组元素

Posted

技术标签:

【中文标题】在训练 SVM 对图像进行分类时设置具有序列错误的数组元素【英文标题】:Setting an array element with a sequence error while training SVM to classify images 【发布时间】:2017-03-23 16:18:25 【问题描述】:

我正在尝试在 python 中使用 SVM 进行图像分类,具有 1022 个图像和 1022 个多类标签(每个标签有 14 个类)。

mypath = 'path'
k = listdir(mypath)
images = np.empty((len(k)-1), dtype=object)
resized_imgs = np.empty((len(k)-1),dtype=object)
for n in range(0, len(k)-1):
    images[n] = cv2.imread(join(mypath,k[n]),0) #Reading images in grayscale
    resized_imgs[n] = cv2.resize(images[n],(32,32)) #Resizing images
for i in range(0,len(k)-1):
    a=resized_imgs[i].mean()
    b=resized_imgs[i].std()
    t=np.ndarray([32,32])
    t.fill(1)
    t=t*a
    resized_imgs[i]=(resized_imgs[i]-a)/b
X_train = resized_imgs

for i in range(len(k)-1):
    X_train[i] = X_train[i].flatten().tolist()

y_train = np.array(y_train)
for i in range(len(k)-1):
    y_train[i] = y_train[i].flatten().tolist()

clf=svm.SVC(gamma=0.001)
clf.fit(X_train,y_train)

现在我得到了错误:

---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
<ipython-input-469-bfab446776df> in <module>()
----> 1 clf.fit(X_train,y_train[:,0])

C:\Users\user\Anaconda3\lib\site-packages\sklearn\svm\base.py in fit(self,  X, y, sample_weight)
    148         self._sparse = sparse and not callable(self.kernel)
    149 
--> 150         X = check_array(X, accept_sparse='csr', dtype=np.float64,       order='C')
    151         y = self._validate_targets(y)
    152 

C:\Users\user\Anaconda3\lib\site-packages\sklearn\utils\validation.py in   check_array(array, accept_sparse, dtype, order, copy, force_all_finite,   ensure_2d, allow_nd, ensure_min_samples, ensure_min_features, warn_on_dtype, estimator)
    371                                       force_all_finite)
    372     else:
--> 373         array = np.array(array, dtype=dtype, order=order, copy=copy)
    374 
    375         if ensure_2d:

ValueError: setting an array element with a sequence.

我不明白这怎么可能是由于 X_train 数组的大小不匹配,因为 X_train 的每个元素都是 1024 个标准化像素值的列表,而 y_train 的每个元素对应于 14 个属性的列表。

任何帮助将不胜感激。谢谢!

【问题讨论】:

【参考方案1】:

我知道我哪里出错了。原来,X_train 的 dtype 是对象类型,所以我不得不将其更改为 float。

同样使用下面的代码,

np.array(list(X_train), dtype=np.float)

【讨论】:

以上是关于在训练 SVM 对图像进行分类时设置具有序列错误的数组元素的主要内容,如果未能解决你的问题,请参考以下文章

Hu矩SVM训练及检测-----OpenCV

Python SVM 设置具有序列错误的数组元素

在 OpenCV 上使用 SVM 训练图像

使用 gist 和 SVM 训练的图像分类

SVM 分类器和测试图像

在 OpencV 中训练 SVM 分类器?