Question

In Python:

detect_edge(im, method) Write a function that takes a gray-scale image and detects edges, with the option of horizontal, vertical or both.

Answer 1

import numpy as np
import argparse
import glob
import cv2

def detect_edge(im,method):

img = cv2.imread(im, cv2.IMREAD_GRAYSCALE)
rows, cols = img.shape

sobel_horizontal = cv2.Sobel(img, cv2.CV_64F, 1, 0, ksize=5)
sobel_vertical = cv2.Sobel(img, cv2.CV_64F, 0, 1, ksize=5)

cv2.imshow('Original', img)
if method = 'horizontal' :
cv2.imshow('Sobel horizontal', sobel_horizontal)
else:
cv2.imshow('Sobel vertical', sobel_vertical)
   
cv2.waitKey(0)