import cv2
import numpy as np
import matplotlib.pyplot as plt
img = cv2.imread('image.jpg', 0)
# Sobel Edge Detection
sobelx = cv2.Sobel(img, cv2.CV_64F, 1, 0, ksize=3)
sobely = cv2.Sobel(img, cv2.CV_64F, 0, 1, ksize=3)
sobel = cv2.magnitude(sobelx, sobely)
# Canny Edge Detection
canny = cv2.Canny(img, 100, 200)
plt.subplot(1,3,1)
plt.imshow(img, cmap='gray')
plt.title('Original')
plt.axis('off')
plt.subplot(1,3,2)
plt.imshow(sobel, cmap='gray')
plt.title('Sobel')
plt.axis('off')
plt.subplot(1,3,3)
plt.imshow(canny, cmap='gray')
plt.title('Canny')
plt.axis('off')
plt.show()