我将计算一些纹理的共同出现矩阵,并在下面看到代码,通常当我在cv2.imread()中硬编码一个地址时,不会出现错误,但现在我使用了一个变量(train_image) ),我得到错误,参数image
必须是一个二维数组'。
train_image=[]
train_images = cvutils.imlist(r"data/lbp/train/")
#n_training_samples = len(train_images)
train_dic = {}
with open('data/lbp/train_label.txt', 'rb') as csvfile:
reader = csv.reader(csvfile, delimiter=',')
for row in reader:
train_dic[row[0]] = int(row[1])
X_train = []
X_name = []
y_train = []
#z_test = []
for train_image in train_images:
im = cv2.imread(train_image)
ngcm= greycomatrix(im, [1], [0], 256, symmetric=False, normed=True)
contrast = greycoprops(ngcm, 'contrast')
im_gray = cv2.cvtColor(im, cv2.COLOR_BGR2GRAY)
radius = 2
no_points = 8 * radius
lbp = local_binary_pattern(im_gray, no_points, radius, method='uniform')
#classifier = svm.SVC(gamma=0.001)
#classifier.fit(data[:n_samples / 2], digits.target[:n_samples / 2])
x = itemfreq(lbp.ravel())
hist = x[:, 1]/sum(x[:, 1])
X_name.append(train_image)
X_train.append(hist)
y_train.append(train_dic[os.path.split(train_image)[1]])
model = LinearSVC(C=100.0, random_state=50)
model.fit(X_train, y_train)
这是我得到的错误:
ValueError Traceback (most recent call last)
<ipython-input-20-9c879ecfb9e6> in <module>()
5 for train_image in train_images:
6 im = cv2.imread(train_image)
----> 7 ngcm= greycomatrix(im, [1], [0], 256, symmetric=False, normed=True)
8 contrast = greycoprops(ngcm, 'contrast')
9 im_gray = cv2.cvtColor(im, cv2.COLOR_BGR2GRAY)
d:\Anaconda2\lib\site-packages\skimage\feature\texture.pyc in greycomatrix(image, distances, angles, levels, symmetric, normed)
90
91 """
---> 92 assert_nD(image, 2)
93 assert_nD(distances, 1, 'distances')
94 assert_nD(angles, 1, 'angles')
d:\Anaconda2\lib\site-packages\skimage\_shared\utils.pyc in assert_nD(array, ndim, arg_name)
163 ndim = [ndim]
164 if not array.ndim in ndim:
--> 165 raise ValueError(msg % (arg_name, '-or-'.join([str(n) for n in ndim])))
ValueError: The parameter `image` must be a 2-dimensional array
感谢任何帮助或提示,谢谢!
答案 0 :(得分:1)
您需要将颜色转换为灰色图像
im = cv2.imread(train_image)
im = cv2.cvtColor(im, cv2.COLOR_BGR2GRAY)
ngcm= greycomatrix(im, [1], [0], 256, symmetric=False, normed=True)
答案 1 :(得分:1)
您需要将图像更改为灰度图像。
如果您使用枕头:
img = io.imread(image, as_gray=True)
如果您使用的是 openCV:
img = cv2.imread(image)
img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)