用python opencv绘制螺旋正方形

import cv2
import numpy as np

image_size = (500, 500)
background_color = (255, 255, 255)  # 白色背景
image = np.ones((image_size[0], image_size[1], 3), dtype=np.uint8) * background_color

center = (image_size[0] // 2, image_size[1] // 2)  # 中心点
square_size = 10  # 正方形边长
gap = 1  # 正方形之间的间隔
rotation_angle = 2  # 每个正方形的旋转角度

for i in range(100):
    side_length = square_size + i * gap
    top_left = (center[0] - side_length // 2, center[1] - side_length // 2)
    bottom_right = (center[0] + side_length // 2, center[1] + side_length // 2)
    cv2.rectangle(image, top_left, bottom_right, (0, 0, 0), 1)
    image = cv2.rotate(image, rotation_angle)

cv2.imshow("Spiral Square", image)
cv2.waitKey(0)
cv2.destroyAllWindows()

cv2.imwrite("spiral_square.png", image)