我需要在Python中进行图像处理。我想使用小波变换作为滤波器组。有人能建议我应该使用哪个库吗?
我已经安装了pywavelet,但我不知道如何与opencv相结合。如果我使用wavedec2命令,则会引发ValueError("Expected 2D input data.")错误。
有人能帮帮我吗?
我需要在Python中进行图像处理。我想使用小波变换作为滤波器组。有人能建议我应该使用哪个库吗?
我已经安装了pywavelet,但我不知道如何与opencv相结合。如果我使用wavedec2命令,则会引发ValueError("Expected 2D input data.")错误。
有人能帮帮我吗?
希望这可以帮助到您。
import numpy as np
import pywt
import cv2
def w2d(img, mode='haar', level=1):
imArray = cv2.imread(img)
#Datatype conversions
#convert to grayscale
imArray = cv2.cvtColor( imArray,cv2.COLOR_RGB2GRAY )
#convert to float
imArray = np.float32(imArray)
imArray /= 255;
# compute coefficients
coeffs=pywt.wavedec2(imArray, mode, level=level)
#Process Coefficients
coeffs_H=list(coeffs)
coeffs_H[0] *= 0;
# reconstruction
imArray_H=pywt.waverec2(coeffs_H, mode);
imArray_H *= 255;
imArray_H = np.uint8(imArray_H)
#Display result
cv2.imshow('image',imArray_H)
cv2.waitKey(0)
cv2.destroyAllWindows()
w2d("test1.png",'db1',10)
cv2.COLOR_BGR2GRAY
。_multilevel.py
的最大级别为 7 而非 10,因此应该这样做:w2d("test1.png",'db1',7)
。