def apply_sobel_edge_detection(image): dx = sobel(image, axis=0, mode='reflect') dy = sobel(image, axis=1, mode='reflect') magnitude = np.sqrt(dx**2 + dy**2) return magnitude