Свертка изображения с ядром дает белый вывод - PullRequest
Новая
       36

Свертка изображения с ядром дает белый вывод

0 голосов
/ 25 января 2019

У меня есть код, который фильтрует изображение с гауссовым ядром 3x3, но вывод белый.Функция GuassianFilter работает (вывод верен), но есть проблема в функции свертки.

В чем проблема?Я снова проверил код, но не смог решить эту проблему. 

import math
import numpy as np
import cv2
path="funny_hats.jpg"
inputImage = cv2.imread(path,cv2.IMREAD_GRAYSCALE)


def GaussianFilter(img): 
    #generating 3x3 kernel
    kernel = np.ones((3,3), dtype='float64')
    size = 3
    mean = int(size/2)
    sigma = 1 # standart deviation is 1
    sumAll = 0
    for i in range(size):
        for j in range(size):
            kernel[i,j] = math.exp(-1* ((math.pow( (i-mean)/sigma, 2.0) + (math.pow((j-mean)/sigma, 2.0)) ) / (2* math.pow(sigma,2)) )) / (sigma * math.pow(2*math.pi, 1/2))
            sumAll += kernel[i,j]
    # normalizing kernel
    for i in range(size):
        for j in range(size):
            kernel[i,j] /= sumAll
    # Filter image with created kernel   
    img = convolution(img, kernel) # filtered image
    print(img)
    cv2.imshow('aa', img)
    cv2.waitKey(0)
    cv2.destroyAllWindows()

def convolution(img, dest):
    res = img
    [h,w] = img.shape
    [kh, kw] = dest.shape # kernel shape
    kr = int(kh/2) # kernel radius
    res = np.zeros(img.shape)

    for i in range(0+kr,h-kr):
        for j in range(0+kr,w-kr):
            for k in range(-1 * kr, kr + 1):
                for m in range(-1 * kr, kr + 1):
                    res[i,j] += dest[k,m]*img[i+k, j+m]
    res[:,0] = res[:, 1]
    res[:,w-1] = res[:, w-2]
    res[0,:] = res[1,:]
    res[h-1,:] = res[h-2,:]
    return res


GaussianFilter(inputImage)

1 Ответ

0 голосов
/ 25 января 2019 
res = img

Это неправильно.Вы должны создать изображение, где все пиксели будут равны нулю (черный).

...