我正在尝试在 python 中的图像上应用 haar 小波。这是代码
from pywt import dwt2, idwt2
img = cv2.imread('xyz.png')
cA, (cH, cV, cD) = dwt2(img, 'haar')
然后我修改嵌入一些数据的系数，如下所示
cH1=cH+k*pn_sequence_h
cV1=cV+k*pn_sequence_v
之后，我使用以下代码应用 idwt
idwt2(cA,(cH1,cV1,cD),'haar')[:Mc,:Nc]
其中 Mc 和 Nc 是分解图像的高度和宽度。
但是，我收到此代码的错误。下面是一个错误。
Traceback (most recent call last):
File "dwt.py", line 15, in <module>
idwt2(cA,(cH1,cV1,cD),'haar')[:Mc,:Nc]
File "C:\Python27\lib\site-packages\pywt\_multidim.py", line 104, in idwt2
LL, (HL, LH, HH) = coeffs
ValueError: too many values to unpack
我应该怎么做才能修复这个错误？我是 python 新手。任何帮助将不胜感激。
我也在下面的代码中尝试过这样的操作。但在这里我没有得到像 CA、CH、CV、CD 这样的系数。我得到的是所有系数。
import numpy as np
import pywt
import numpy
import PIL
from PIL import Image
img = PIL.Image.open("rot.png").convert("L")
imgarr = numpy.array(img)
coeffs = pywt.dwt2(imgarr, 'haar')
pywt.idwt2(coeffs, 'haar')