| import cv2 | |
| def sobel_cam(img): | |
| gray = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY) | |
| grad_x = cv2.Sobel(gray, cv2.CV_64F, 1, 0, ksize=3) | |
| grad_y = cv2.Sobel(gray, cv2.CV_64F, 0, 1, ksize=3) | |
| abs_grad_x = cv2.convertScaleAbs(grad_x) | |
| abs_grad_y = cv2.convertScaleAbs(grad_y) | |
| grad = cv2.addWeighted(abs_grad_x, 0.5, abs_grad_y, 0.5, 0) | |
| return grad | |