在Python中使用opencv对图片头像加个正方形

在Python中，我们可以利用OpenCV库（Open Source Computer Vision Library）来处理图像，包括添加边框，比如正方形。以下是一个简单的步骤说明：

首先，你需要安装cv2模块，如果你还没有安装，可以使用pip来安装：

pip install opencv-python

然后，你可以使用以下代码片段来加载图片、添加正方形边框，并保存修改后的图片：

import cv2
import numpy as np

# 加载图片
img = cv2.imread('path_to_your_image.jpg')

# 定义边框大小和颜色
square_size = 50
color = (0, 255, 0)  # 绿色边框，BGR格式

# 计算需要添加边框的新尺寸
height, width = img.shape[:2]
new_width = max(width + square_size, height + square_size)
new_height = new_width

# 创建一个新的全黑背景，大小等于新尺寸
new_img = np.zeros((new_height, new_width, 3), dtype=np.uint8)

# 将原始图片复制到新图的中心位置
x_start = (new_width - width) // 2
y_start = (new_height - height) // 2
new_img[y_start:y_start+height, x_start:x_start+width] = img

# 添加正方形边框
border_thickness = 2
cv2.rectangle(new_img, (x_start, y_start), (x_start+width, y_start+height), color, border_thickness)

# 保存结果
cv2.imwrite('output_image_with_square.jpg', new_img)