如何将小波变换的LL,LH,HL保存为3通道图像?

时间:2019-05-14 13:38:55

标签: python wavelet-transform

试图使用此link中描述的PyWavelets进行小波变换,这是我的代码:

import pywt
import cv2 as cv
img = cv.imread('D:/gray.jpeg', cv.IMREAD_GRAYSCALE) #read gray image.
img = cv.resize(img, (256,256))
coeffs2 = pywt.dwt2(img, 'bior1.3')
LL, (LH, HL, HH) = coeffs2

现在,我想将LL,LH和HH保存为3通道图像,但是我不知道该怎么做,有人可以告诉我该怎么做吗?

我的另一个问题:这仅适用于灰度图像吗?

1 个答案:

答案 0 :(得分:0)

将彩色图像分成三个颜色通道

b,g,r = cv2.split(img)

coeffsb = pywt.dwt2(b, 'bior1.3')
LLb, (LHb, HLb, HHb) = coeffsb

coeffsg = pywt.dwt2(g, 'bior1.3')
LLg, (LHg, HLg, HHg) = coeffsg

coeffsb = pywt.dwt2(b, 'bior1.3')
LLr, (LHr, HLr, HHb) = coeffsr

然后您可以分别对每个通道应用小波变换。之后,您可以合并您选择的频段(例如 LL):

LL = cv2.merge(LLb, LLg, LLr)