Python的ValueError异常在此代码

-3

File "project.py", line 35, in <module> 
    cells = [np.hsplit(row,100) for row in np.vsplit(img,50)] 
    File "/usr/local/lib/python2.7/site-packages/numpy/lib/shape_base.py", line 623, in vsplit 
    return split(ary, indices_or_sections, 0) 
    File "/usr/local/lib/python2.7/site-packages/numpy/lib/shape_base.py", line 508, in split 
    'array split does not result in an equal division') 
ValueError: array split does not result in an equal division

代码

import cv2 
import numpy as np 

SZ=20 
bin_n = 16 # Number of bins 

svm_params = dict(kernel_type = cv2.SVM_LINEAR, 
        svm_type = cv2.SVM_C_SVC, 
        C=2.67, gamma=5.383) 

affine_flags = cv2.WARP_INVERSE_MAP|cv2.INTER_LINEAR 

def deskew(img): 
    m = cv2.moments(img) 
    if abs(m['mu02']) < 1e-2: 
     return img.copy() 
    skew = m['mu11']/m['mu02'] 
    M = np.float32([[1, skew, -0.5*SZ*skew], [0, 1, 0]]) 
    img = cv2.warpAffine(img,M,(SZ, SZ),flags=affine_flags) 
    return img 

def hog(img): 
    gx = cv2.Sobel(img, cv2.CV_32F, 1, 0) 
    gy = cv2.Sobel(img, cv2.CV_32F, 0, 1) 
    mag, ang = cv2.cartToPolar(gx, gy) 
    bins = np.int32(bin_n*ang/(2*np.pi)) # quantizing binvalues in (0...16) 
    bin_cells = bins[:10,:10], bins[10:,:10], bins[:10,10:], bins[10:,10:] 
    mag_cells = mag[:10,:10], mag[10:,:10], mag[:10,10:], mag[10:,10:] 
    hists = [np.bincount(b.ravel(), m.ravel(), bin_n) for b, m in zip(bin_cells, mag_cells)] 
    hist = np.hstack(hists)  # hist is a 64 bit vector 
    return hist 

img = cv2.imread('digits.png',0) 

cells = [np.hsplit(row,100) for row in np.vsplit(img,50)] 

# First half is trainData, remaining is testData 
train_cells = [ i[:50] for i in cells ] 
test_cells = [ i[50:] for i in cells] 

######  Now training  ######################## 

deskewed = [map(deskew,row) for row in train_cells] 
hogdata = [map(hog,row) for row in deskewed] 
trainData = np.float32(hogdata).reshape(-1,64) 
responses = np.float32(np.repeat(np.arange(10),250)[:,np.newaxis]) 

svm = cv2.SVM() 
svm.train(trainData,responses, params=svm_params) 
svm.save('svm_data.dat') 

######  Now testing  ######################## 

deskewed = [map(deskew,row) for row in test_cells] 
hogdata = [map(hog,row) for row in deskewed] 
testData = np.float32(hogdata).reshape(-1,bin_n*4) 
result = svm.predict_all(testData) 

####### Check Accuracy ######################## 
mask = result==responses 
correct = np.count_nonzero(mask) 
print correct*100.0/result.size

来源

2017-04-21 SOWMITHRA KUMAR G M

欢迎来到SO。你必须显示你的代码，所以我们可以帮助你 – mbatchkarov

非常难以帮助错误，当我们不能看到什么导致它 – WhatsThePoint

你可能试图打破载体导入线。但是，请确保矢量元素的数量是您试图将其分解成的行数的倍数。例如，不能将（5,1）矢量分成两行。在这种情况下，每个row的尺寸必须为100

来源

2017-04-21 10:18:55

从错误的倍数和搜索的代码（用于`分割）必须发生错误在

[np.hsplit(row,100) for row in np.vsplit(img,50)]

什么形状img？该错误表示它（img.shape[0]）不是50的倍数。

来源

2017-04-21 15:59:54 hpaulj

Python的ValueError异常在此代码

回答

相关问题