这里写自定义目录标题
import cv2
import numpy as np
def simulate_polarizer(image, angle_degrees, intensity=0.5):
gray_image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
blurred_image = cv2.GaussianBlur(gray_image, (5, 5), 0)
sobel_x = cv2.Sobel(blurred_image, cv2.CV_64F, 1, 0, ksize=3)
sobel_y = cv2.Sobel(blurred_image, cv2.CV_64F, 0, 1, ksize=3)
gradient_magnitude = np.sqrt(sobel_x ** 2 + sobel_y ** 2)
gradient_direction = np.arctan2(sobel_y, sobel_x)
angle_radians = np.radians(angle_degrees)
angle_difference = np.abs(gradient_direction - angle_radians)
angle_difference = np.minimum(angle_difference, np.pi - angle_difference)
polarization_effect = 1 - intensity * np.cos(2 * angle_difference)
polarized_image = cv2.merge([gray_image * polarization_effect] * 3)
return polarized_image.astype(np.uint8)
image = cv2.imread(‘input_image.jpg’)
polarized_image = simulate_polarizer(image, angle_degrees=45, intensity=0.5)
cv2.imshow(‘Original Image’, image)
cv2.imshow(‘Polarized Image’, polarized_image)
cv2.waitKey(0)
cv2.destroyAllWindows()