Как совместить pyWavelet и openCV для обработки изображений?

Мне нужно сделать обработку изображений в Python. Я хочу использовать вейвлет-преобразование в качестве банка фильтров. Может кто-нибудь предложить мне, какую библиотеку я должен использовать? У меня был установлен pywavelet, но я не знаю, как совместить его с opencv. Если я использую команду wavedec2, она вызывает ValueError("Ожидаемые двумерные входные данные").

Может кто-нибудь мне помочь?

Источник

2 ответа

Решение

Надеюсь это поможет

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)
Источник

Ответ Наванита правильный, но с двумя исправлениями:

1- OpenCV читать и сохранять изображения как BGR, а не RGB, так что вы должны сделать cv2.COLOR_BGR2GRAY если быть точным.

2- Максимальный уровень _multilevel.py это 7, а не 10, поэтому вы должны сделать: w2d("test1.png",'db1',7)

Источник
Другие вопросы по тегам