程序运行提示ValueError: Buffer has wrong number of dimensions (expected 2, got 3),程序如下img = cv2.imread('gaussian.bmp') skel, distance = morphology.medial_axis(img, return_distance=True) # 图片细化(骨架提取) dist_on_skel = distance * skel cv2.imwrite("skeleton_gaussian_1.bmp", dist_on_skel)
时间: 2024-03-21 15:42:06 浏览: 123
python3.7解决最小二乘遇到ValueError:Expected 2D array, got 1D array instead: array=[5.].关于reshape和predict
这个错误通常是由于cv2.imread函数读取的图像是一个三维的numpy数组,而morphology.medial_axis函数期望的是一个二维的二值图像。因此,需要对读取进来的图像进行预处理,将其转换为一个二维的二值图像,才能够传递给morphology.medial_axis函数进行处理。
对于这种情况,可以尝试将读取进来的图像转换为灰度图像,然后进行二值化处理,再进行骨架提取操作。代码如下:
```
import cv2
import numpy as np
from skimage import morphology
# 读取图像并转换为灰度图像
img = cv2.imread('gaussian.bmp', cv2.IMREAD_GRAYSCALE)
# 二值化图像
_, img_bw = cv2.threshold(img, 0, 255, cv2.THRESH_BINARY_INV + cv2.THRESH_OTSU)
# 图像细化(骨架提取)
skel, distance = morphology.medial_axis(img_bw, return_distance=True)
# 计算每个像素到骨架的距离
dist_on_skel = distance * skel
# 将结果转换为图像并保存
cv2.imwrite("skeleton_gaussian_1.bmp", dist_on_skel.astype(np.uint8) * 255)
```
在这段代码中,首先使用cv2.imread函数读取图像,并将其转换为灰度图像。然后,使用cv2.threshold函数对图像进行二值化处理,并生成一个二维的二值图像。最后,将二值图像传递给morphology.medial_axis函数进行骨架提取操作,得到每个像素到骨架的距离信息,并将距离信息转换为图像格式并保存。
阅读全文