Prewits_y.py

import numpy as np
import cv2
from matplotlib import pyplot as plt
img = cv2.imread("imgs/chapter5/sudoku.png", 0);
kernel = [
            [1, 1, 1],
            [0, 0, 0],
            [-1, -1, -1]
]

kernel = np.array(kernel); #Normalized mean kernel
prewit = cv2.filter2D(img, -1, kernel);

f = plt.figure(figsize=(15,15))
f.add_subplot(1, 2, 1).set_title('Original Image');
plt.imshow(img, cmap = "gray")
f.add_subplot(1, 2, 2).set_title('Prewitt in y direction Image');
plt.imshow(prewit, cmap = "gray");
plt.show()