如何结合 pyWavelet 和 openCV 进行图像处理？答案

【问题标题】：How to Combine pyWavelet and openCV for image processing?如何结合 pyWavelet 和 openCV 进行图像处理？
【发布时间】：2014-07-02 16:36:50
【问题描述】：

我需要在 python 中进行图像处理。我想使用小波变换作为滤波器组。谁能建议我应该使用哪个库？我安装了 pywavelet，但我不知道如何将它与 opencv 结合使用。如果我使用 wavedec2 命令，它会引发 ValueError("Expected 2D input data.")

谁能帮帮我？

【问题讨论】：

标签： python opencv image-processing dwt

【解决方案1】：

希望对你有帮助

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)

【讨论】：

【解决方案2】：

Navaneeth 的答案是正确的，但有两个更正：

1- Opencv 读取图像并将其保存为 BGR 而不是 RGB，因此准确地说，您应该使用 cv2.COLOR_BGR2GRAY。

2- _multilevel.py 的最大级别是 7 而不是 10，所以你应该这样做：w2d("test1.png",'db1',7)

【讨论】：