Как расширить pyWavelets для работы с N-мерными данными?

Это может быть вопрос для другого форума, если да, пожалуйста, дайте мне знать. Я заметил, что только 14 человек следуют вейвлет-тегу.

У меня здесь элегантный способ расширения вейвлет-декомпозиции в pywt (pywavelets package) до нескольких измерений. Это должно закончиться из коробки, если pywt установлен. Тест 1 показывает декомпозицию и перекомпозицию 3D-массива. Все, что нужно сделать, это увеличить количество измерений, и код будет работать при разложении / перекомпозиции с 4, 6 или даже 18 размеры данных.

Я заменил pywt.wavedec и pywt.здесь функционирует waverec. Кроме того, в fn_dec я показываю, как новая функция wavedec работает так же, как и старая.

есть один улов: он представляет вейвлет-коэффициенты в виде массива той же формы, что и данные. Как следствие, с моими ограниченными знаниями вейвлетов, я смог использовать его только для вейвлетов Хаара. Другие подобные ДБ4 например стравить коэффициентов по краям этих строгих границ (не проблема с текущим представлением коэффициентов в виде списка массивов [CA, CD1 ... CDN]. Еще одна загвоздка в том, что я работал только с 2^N реберными кубоидами данных.

теоретически, я думаю, что должно быть возможно убедиться, что" кровотечение " не происходит. Алгоритм для такого рода вейвлет-декомпозиции и рекомпозиции обсуждается в "численные рецепты в C" - Уильям брюк, Саул в teukolsky, Уильям Т. Vetterling и Брайан П. Фланнери (Второе Издание). Хотя этот алгоритм предполагает отражение на ребрах, а не другие формы расширений ребер (например, zpd), метод достаточно общий для работы с другими формами расширения.

любые предложения о том, как распространить эту работу на другие вейвлеты?

Примечание: этот запрос также размещен наhttp://groups.google.com/group/pywavelets

спасибо, Ajo

import pywt
import sys
import numpy as np

def waveFn(wavelet):
    if not isinstance(wavelet, pywt.Wavelet):
        return pywt.Wavelet(wavelet)
    else:
        return wavelet

# given a single dimensional array ... returns the coefficients.
def wavedec(data, wavelet, mode='sym'):
    wavelet = waveFn(wavelet)

    dLen = len(data)
    coeffs = np.zeros_like(data)
    level = pywt.dwt_max_level(dLen, wavelet.dec_len)

    a = data    
    end_idx = dLen
    for idx in xrange(level):
        a, d = pywt.dwt(a, wavelet, mode)
        begin_idx = end_idx/2
        coeffs[begin_idx:end_idx] = d
        end_idx = begin_idx

    coeffs[:end_idx] = a
    return coeffs

def waverec(data, wavelet, mode='sym'):
    wavelet = waveFn(wavelet)

    dLen = len(data)
    level = pywt.dwt_max_level(dLen, wavelet.dec_len)

    end_idx = 1
    a = data[:end_idx] # approximation ... also the original data 
    d = data[end_idx:end_idx*2]    
    for idx in xrange(level):
        a = pywt.idwt(a, d, wavelet, mode)
        end_idx *= 2
        d = data[end_idx:end_idx*2]
    return a

def fn_dec(arr):
    return np.array(map(lambda row: reduce(lambda x,y : np.hstack((x,y)), pywt.wavedec(row, 'haar', 'zpd')), arr))
    # return np.array(map(lambda row: row*2, arr))

if __name__ == '__main__':
    test  = 1
    np.random.seed(10)
    wavelet = waveFn('haar')
    if test==0:
        # SIngle dimensional test.
        a = np.random.randn(1,8)
        print "original values A"
        print a
        print "decomposition of A by method in pywt"
        print fn_dec(a)
        print " decomposition of A by my method"
        coeffs =  wavedec(a[0], 'haar', 'zpd')
        print coeffs
        print "recomposition of A by my method"
        print waverec(coeffs, 'haar', 'zpd')
        sys.exit()
    if test==1:
        a = np.random.randn(4,4,4)
        # 2 D test
        print "original value of A"
        print a

        # decompose the signal into wavelet coefficients.
        dimensions = a.shape
        for dim in dimensions:
            a = np.rollaxis(a, 0, a.ndim)
            ndim = a.shape
            #a = fn_dec(a.reshape(-1, dim))
            a = np.array(map(lambda row: wavedec(row, wavelet), a.reshape(-1, dim)))
            a = a.reshape(ndim)
        print " decomposition of signal into coefficients"
        print a

        # re-composition of the coefficients into original signal
        for dim in dimensions:
            a = np.rollaxis(a, 0, a.ndim)
            ndim = a.shape
            a = np.array(map(lambda row: waverec(row, wavelet), a.reshape(-1, dim)))
            a = a.reshape(ndim)
        print "recomposition of coefficients to signal"
        print a