7

Pythonで画像処理を行う必要があります。フィルターバンクとしてウェーブレット変換を使用したい。どのライブラリを使用すればよいか、誰かに提案してもらえますか? pywavelet をインストールしましたが、opencv と組み合わせる方法がわかりません。wavedec2 コマンドを使用すると、ValueError("Expected 2D input data.") が発生します。

誰でも私を助けることができますか？

4

2 に答える 2

15

お役に立てれば

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)
于 2014-07-14T23:39:25.727 に答える
5

Navaneethの回答は正しいですが、2 つの修正があります。

1-Opencvは画像をRGBではなくBGRとして読み取って保存するため、cv2.COLOR_BGR2GRAY正確に行う必要があります。

2- の最大レベルは_multilevel.py10 ではなく 7 であるため、次のことを行う必要があります。w2d("test1.png",'db1',7)

于 2017-07-21T14:55:59.600 に答える