3Dnumpy阵列错误中的单级未抽取小波变换

Question

我一直在尝试使用python中的pywavelets包中的swtn函数将一级非抽取小波变换应用到3D numpy排列中，如下所示：

import numpy as np
from pywavelts import swtn

img = np.random.rand(4,4,5)
WT = swtn(img, 'coif1', level = 1, start_level = 0) 

这会引发一个错误：

Traceback (most recent call last):

  File "<ipython-input-60-1f6f4144a239>", line 1, in <module>
    pywt.swtn(img, 'coif1', level=1, start_level =0, axes=None)

  File "/usr/local/lib/python2.7/dist-packages/pywt/_swt.py", line 387, in swtn
    axis=axis)[0]

  File "pywt/_extensions/_swt.pyx", line 100, in pywt._extensions._swt.swt_axis (pywt/_extensions/_swt.c:6203)

  File "pywt/_extensions/_swt.pyx", line 116, in pywt._extensions._swt.swt_axis (pywt/_extensions/_swt.c:5035)

ValueError: start_level must be less than 0. 

然而，这对我来说没有意义，因为小波分解不允许起始级别值小于0。有没有人能帮我看一下，指出我漏掉了什么？

Answer 1

我刚刚也遇到了同样的错误。这是由于您使用的数组的大小:因为它不能被2整除，所以函数给出的最大层数：

pywt.swt_max_level

是0。例如，如下所示：

img = np.random.rand(4,4,6)
WT = swtn(img, 'coif1', level = 1, start_level = 0) 

这对我很有效。

Answer 2

摘自pywt.swtn documentation page上的“笔记”

...要求沿变换轴的信号长度是2**电平的倍数。如果不是这样，用户应该使用诸如numpy.pad之类的函数填充到合适的大小。

在您的例子中，第三个轴的尺寸应该是2的倍数。